diff --git a/graph_net/subgraph_decompose_and_evaluation_step.py b/graph_net/subgraph_decompose_and_evaluation_step.py
index c14f6b369..6a1d1b437 100755
--- a/graph_net/subgraph_decompose_and_evaluation_step.py
+++ b/graph_net/subgraph_decompose_and_evaluation_step.py
@@ -415,7 +415,13 @@ def run_evaluation(
             work_dir, "reference_device_outputs"
         )
 
-    cmd = [sys.executable, "-m", f"graph_net.{framework}.{test_module_name}"] + [
+    test_module_path = (
+        f"graph_net_bench.{framework}.{test_module_name}"
+        if test_module_name == "test_compiler"
+        else f"graph_net.{framework}.{test_module_name}"
+    )
+
+    cmd = [sys.executable, "-m", test_module_path] + [
         item
         for key, value in test_module_arguments.items()
         for item in (f"--{key}", str(value))
diff --git a/graph_net/torch/test_reference_device.py b/graph_net/torch/test_reference_device.py
index f022d2ba5..46f5446ee 100644
--- a/graph_net/torch/test_reference_device.py
+++ b/graph_net/torch/test_reference_device.py
@@ -137,7 +137,7 @@ def test_multi_models(args):
 def main(args):
     assert os.path.isdir(args.model_path)
     # Support all torch compilers
-    valid_compilers = list(test_compiler.registry_backend.keys())
+    valid_compilers = list(test_compiler.compiler_backend_name2class.keys())
     assert (
         args.compiler in valid_compilers
     ), f"Compiler must be one of {valid_compilers}"
diff --git a/graph_net/torch/test_target_device.py b/graph_net/torch/test_target_device.py
index ec2085a32..87fe22077 100644
--- a/graph_net/torch/test_target_device.py
+++ b/graph_net/torch/test_target_device.py
@@ -8,7 +8,8 @@
 from graph_net_bench import path_utils
 from graph_net_bench import test_compiler_util
 from graph_net import model_path_util
-from graph_net_bench.torch import test_compiler, test_reference_device
+from graph_net_bench.torch import test_compiler
+from graph_net.torch import test_reference_device
 
 
 def parse_config_from_reference_log(log_path):