Index: tools/telemetry/telemetry/page/record_wpr.py |
diff --git a/tools/telemetry/telemetry/page/record_wpr.py b/tools/telemetry/telemetry/page/record_wpr.py |
index 6936f4529b9ed6169f30b1b5f6176be75b51ff3b..bbef6fb5eb8dcbd067e645c2f09e11dad3b5d127 100755 |
--- a/tools/telemetry/telemetry/page/record_wpr.py |
+++ b/tools/telemetry/telemetry/page/record_wpr.py |
@@ -8,7 +8,7 @@ import sys |
import tempfile |
import time |
-from telemetry import test |
+from telemetry import benchmark |
from telemetry.core import browser_options |
from telemetry.core import discover |
from telemetry.core import wpr_modes |
@@ -88,7 +88,7 @@ def Main(base_dir): |
# Filter out unneeded ProfileCreators (crbug.com/319573). |
if not issubclass(cls, profile_creator.ProfileCreator) |
} |
- tests = discover.DiscoverClasses(base_dir, base_dir, test.Test, |
+ tests = discover.DiscoverClasses(base_dir, base_dir, benchmark.Benchmark, |
index_by_class_name=True) |
options = browser_options.BrowserFinderOptions() |