diff --git a/test/test_fmm.py b/test/test_fmm.py
index c1df96a6d65ec147eb523d1326b887f8521aeae8..179994d5f91dcb6c7b7c7a5fafb9d207d6401a25 100644
--- a/test/test_fmm.py
+++ b/test/test_fmm.py
@@ -210,7 +210,7 @@ def test_sumpy_fmm(ctx_getter, knl, local_expn_class, mpole_expn_class):
                 partial(local_expn_class, knl),
                 out_kernels)
         wrangler = wcc.get_wrangler(queue, tree, dtype,
-                fmm_level_to_order=lambda lev: order,
+                fmm_level_to_order=lambda tree, lev: order,
                 kernel_extra_kwargs=extra_kwargs)
 
         from boxtree.fmm import drive_fmm
@@ -283,7 +283,7 @@ def test_sumpy_fmm_exclude_self(ctx_getter):
             exclude_self=True)
 
     wrangler = wcc.get_wrangler(queue, tree, dtype,
-            fmm_level_to_order=lambda lev: order,
+            fmm_level_to_order=lambda tree, lev: order,
             self_extra_kwargs=self_extra_kwargs)
 
     from boxtree.fmm import drive_fmm