diff --git a/test/test_integration.py b/test/test_integration.py
index d0e1cd25b..a5b2d0a25 100644
--- a/test/test_integration.py
+++ b/test/test_integration.py
@@ -57,13 +57,13 @@ def wait_for_element_by_css_selector(self, selector):
         )
 
     def wait_for_text_to_equal(self, selector, assertion_text):
+
+        def text_equal(driver):
+            text = driver.find_element_by_css_selector(selector).text
+            return text == assertion_text
+
         WebDriverWait(self.driver, TIMEOUT).until(
-            EC.text_to_be_present_in_element((By.CSS_SELECTOR, selector),
-                                             assertion_text)
-        )
-        self.assertEqual(
-            assertion_text,
-            self.driver.find_element_by_css_selector(selector).text
+            text_equal
         )
 
     def snapshot(self, name):