Index: tools/chrome_proxy/live_tests/pagesets/__init__.py |
diff --git a/tools/chrome_proxy/live_tests/pagesets/__init__.py b/tools/chrome_proxy/live_tests/pagesets/__init__.py |
index 1ea4db1ff48cc8b604a3dff8c8f8fccbf931f5cc..45ce4af9df80c564b044b526df9669c00e3a809c 100644 |
--- a/tools/chrome_proxy/live_tests/pagesets/__init__.py |
+++ b/tools/chrome_proxy/live_tests/pagesets/__init__.py |
@@ -7,13 +7,13 @@ import os |
import sys |
from telemetry.core import discover |
-from telemetry.page import page_set |
+from telemetry import story |
-# Import all submodules' PageSet classes. |
+# Import all submodules' StorySet classes. |
start_dir = os.path.dirname(os.path.abspath(__file__)) |
top_level_dir = os.path.abspath(os.path.join(start_dir, os.pardir, os.pardir)) |
-base_class = page_set.PageSet |
+base_class = story.StorySet |
for cls in discover.DiscoverClasses( |
start_dir, top_level_dir, base_class).values(): |
setattr(sys.modules[__name__], cls.__name__, cls) |