1 files changed, 2 insertions, 4 deletions
diff --git a/systrace/catapult/telemetry/telemetry/internal/backends/chrome/cros_test_case.py b/systrace/catapult/telemetry/telemetry/internal/backends/chrome/cros_test_case.py
index 9100d7a..259157b 100644
--- a/systrace/catapult/telemetry/telemetry/internal/backends/chrome/cros_test_case.py
+++ b/systrace/catapult/telemetry/telemetry/internal/backends/chrome/cros_test_case.py
@@ -11,8 +11,6 @@ from telemetry.internal.browser import browser_finder
 from telemetry.internal.browser import extension_to_load
 from telemetry.testing import options_for_unittests
 
-import py_utils
-
 
 class CrOSTestCase(unittest.TestCase):
   def setUp(self):
@@ -79,5 +77,5 @@ class CrOSTestCase(unittest.TestCase):
           window.__login_status = s;
         });
     ''')
-    return py_utils.WaitFor(
-        lambda: extension.EvaluateJavaScript('window.__login_status'), 10)
+    return extension.WaitForJavaScriptCondition(
+        'window.__login_status', timeout=10)