diff --git a/modules/benchmarks/src/class_bindings/BUILD.bazel b/modules/benchmarks/src/class_bindings/BUILD.bazel
index 48c3cdec70..dbf1d0b038 100644
--- a/modules/benchmarks/src/class_bindings/BUILD.bazel
+++ b/modules/benchmarks/src/class_bindings/BUILD.bazel
@@ -1,7 +1,7 @@
 package(default_visibility = ["//modules/benchmarks:__subpackages__"])
 
 load("@io_bazel_rules_sass//sass:sass.bzl", "sass_binary")
-load("//dev-infra/benchmark/component_benchmark:component_benchmark.bzl", "component_benchmark")
+load("//tools/components:component_benchmark.bzl", "component_benchmark")
 
 sass_binary(
     name = "class_bindings_styles",
@@ -12,8 +12,7 @@ component_benchmark(
     name = "benchmark",
     driver = ":class_bindings.perf-spec.ts",
     driver_deps = [
-        "//dev-infra/benchmark/driver-utilities",
-        "@npm//@types/jasmine",
+        "//modules/e2e_util",
         "@npm//protractor",
     ],
     ng_deps = [