diff --git a/python/tests/test_array_api.py b/python/tests/test_array_api.py
index ec69ebc..9017ca6 100644
--- a/python/tests/test_array_api.py
+++ b/python/tests/test_array_api.py
@@ -412,7 +412,7 @@ def test_optimize(self, snapshot_py, benchmark):
         y = NDArray.var("y")
         with egraph.set_current():
             expr = lda(X, y)
-            simplified = benchmark(simplify_lda, egraph, expr)
+            simplified = benchmark.pendatic(simplify_lda, args=(egraph, expr))
         assert str(simplified) == snapshot_py
 
     # @pytest.mark.xfail(reason="Original source is not working")