diff --git a/test/test_fmm.py b/test/test_fmm.py
index f7bf0540941cc29c1be38113f14afa25e5e32436..73b80ef05d0e0d8c97721faae827765b5e23136b 100644
--- a/test/test_fmm.py
+++ b/test/test_fmm.py
@@ -36,6 +36,7 @@ from sumpy.expansion.local import (
     VolumeTaylorLocalExpansion,
     H2DLocalExpansion, Y2DLocalExpansion,
     LinearPDEConformingVolumeTaylorLocalExpansion)
+from sumpy.fmm import SumpyTranslationClassesData
 
 import pytest
 import warnings
@@ -177,7 +178,6 @@ def test_sumpy_fmm(ctx_factory, knl, local_expn_class, mpole_expn_class,
         target_kernels = [knl]
 
         from sumpy.fmm import (SumpyExpansionWranglerCodeContainer,
-            SumpyTranslationClassesData,
             SumpyTranslationClassesDataNotSuppliedWarning)
 
         if optimized_m2l:
@@ -301,7 +301,8 @@ def test_unified_single_and_double(ctx_factory):
                 strength_usage=strength_usage)
         wrangler = wcc.get_wrangler(queue, tree, dtype,
                 fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
-                source_extra_kwargs=source_extra_kwargs)
+                source_extra_kwargs=source_extra_kwargs,
+                translation_classes_data=SumpyTranslationClassesData(queue, trav))
 
         from boxtree.fmm import drive_fmm
 
@@ -362,7 +363,8 @@ def test_sumpy_fmm_timing_data_collection(ctx_factory):
             target_kernels)
 
     wrangler = wcc.get_wrangler(queue, tree, dtype,
-            fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order)
+            fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
+            translation_classes_data=SumpyTranslationClassesData(queue, trav))
     from boxtree.fmm import drive_fmm
 
     timing_data = {}
@@ -421,7 +423,8 @@ def test_sumpy_fmm_exclude_self(ctx_factory):
 
     wrangler = wcc.get_wrangler(queue, tree, dtype,
             fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
-            self_extra_kwargs=self_extra_kwargs)
+            self_extra_kwargs=self_extra_kwargs,
+            translation_classes_data=SumpyTranslationClassesData(queue, trav))
 
     from boxtree.fmm import drive_fmm
 
@@ -496,7 +499,8 @@ def test_sumpy_axis_source_derivative(ctx_factory):
 
         wrangler = wcc.get_wrangler(queue, tree, dtype,
                 fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
-                self_extra_kwargs=self_extra_kwargs)
+                self_extra_kwargs=self_extra_kwargs,
+                translation_classes_data=SumpyTranslationClassesData(queue, trav))
 
         from boxtree.fmm import drive_fmm
 
@@ -562,7 +566,8 @@ def test_sumpy_target_point_multiplier(ctx_factory):
 
     wrangler = wcc.get_wrangler(queue, tree, dtype,
             fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
-            self_extra_kwargs=self_extra_kwargs)
+            self_extra_kwargs=self_extra_kwargs,
+            translation_classes_data=SumpyTranslationClassesData(queue, trav))
 
     from boxtree.fmm import drive_fmm