diff --git a/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py b/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py
index 7154bc518..a0c97b6a1 100644
--- a/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py
+++ b/tests/test_models/test_mutables/test_units/test_mutable_channel_units.py
@@ -121,7 +121,7 @@ def test_replace_with_dynamic_ops(self):
             for unit_type in GROUPS:
                 with self.subTest(model=model_data, unit=unit_type):
                     model: nn.Module = model_data()
-                    graph = ModuleGraph.init_from_fx_tracer(model)
+                    graph = ModuleGraph.init_from_backward_tracer(model)
                     units: List[
                         MutableChannelUnit] = unit_type.init_from_graph(
                             graph)