diff --git a/tritonbench/components/do_bench/run.py b/tritonbench/components/do_bench/run.py
index e93f37d5..29acba7c 100644
--- a/tritonbench/components/do_bench/run.py
+++ b/tritonbench/components/do_bench/run.py
@@ -475,7 +475,7 @@ def do_bench_wrapper(
                 if latency_measure_mode == "profiler"
                 else _do_bench_inductor
                 if latency_measure_mode == "inductor_benchmarker"
-                else triton.testing.do_bench
+                else triton.runtime.driver.active.get_benchmarker()
             )
 
             return Latency(