Index: tools/perf/page_sets/__init__.py |
diff --git a/tools/perf/page_sets/__init__.py b/tools/perf/page_sets/__init__.py |
index 7504b80dcbf109193bae7ed8435e17a907ae1705..e6a166f69773fc570c8126f6405bd7a4d9fdcadc 100644 |
--- a/tools/perf/page_sets/__init__.py |
+++ b/tools/perf/page_sets/__init__.py |
@@ -8,12 +8,13 @@ import sys |
from telemetry.core import discover |
from telemetry import story |
+from telemetry.story import expectations |
# Import all submodules' PageSet classes. |
start_dir = os.path.dirname(os.path.abspath(__file__)) |
top_level_dir = os.path.dirname(start_dir) |
-base_classes = [story.StorySet, story.expectations.StoryExpectations] |
+base_classes = [story.StorySet, expectations.StoryExpectations] |
for base_class in base_classes: |
for cls in discover.DiscoverClasses( |