diff --git a/asv_bench/benchmarks/finite_difference_1D.py b/asv_bench/benchmarks/finite_difference_1D.py
index d56b67e..e791cef 100644
--- a/asv_bench/benchmarks/finite_difference_1D.py
+++ b/asv_bench/benchmarks/finite_difference_1D.py
@@ -15,7 +15,8 @@ def setup(self):
     def time_derivative(self):
         self.differentiator.d(X=self.t, t=self.t, axis=0)
 
+    def mem_derivative(self):
+        self.differentiator.d(X=self.t, t=self.t, axis=0)
 
-class MemSuite:
-    def mem_list(self):
-        return [0] * 256
+    def peakmem_derivative(self):
+        self.differentiator.d(X=self.t, t=self.t, axis=0)