Index: tools/telemetry/telemetry/core/tab_unittest.py |
diff --git a/tools/telemetry/telemetry/core/tab_unittest.py b/tools/telemetry/telemetry/core/tab_unittest.py |
index 3cc3cd08164d31d510804129e72ae8998eca8811..3bf8a64cbef08a06278a34fba94351e4e3d98d86 100644 |
--- a/tools/telemetry/telemetry/core/tab_unittest.py |
+++ b/tools/telemetry/telemetry/core/tab_unittest.py |
@@ -4,7 +4,7 @@ |
import logging |
-from telemetry import test |
+from telemetry import benchmark |
from telemetry.core import bitmap |
from telemetry.core import video |
from telemetry.core import util |
@@ -158,7 +158,7 @@ class GpuTabTest(tab_test_case.TabTestCase): |
super(GpuTabTest, self).setUp() |
# Test flaky on mac: http://crbug.com/358664 |
- @test.Disabled('android', 'mac') |
+ @benchmark.Disabled('android', 'mac') |
def testScreenshot(self): |
if not self._tab.screenshot_supported: |
logging.warning('Browser does not support screenshots, skipping test.') |