fix(browser): wait.

2e0byo · Jan 29, 2022 · 6873228 · 6873228
1 parent 9ed5618
commit 6873228
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yadc/browser.py b/yadc/browser.py
@@ -128,10 +128,10 @@ def bypass(self):
     def wait(self):
         waited = 0
         self._logger.debug("Waiting as page is not ready yet.")
-        if self.WAIT_MSG in self._driver.page_source() and waited < self.MAX_WAIT:
+        if self.WAIT_MSG in self.page_source and waited < self.MAX_WAIT:
             sleep(1)
             waited += 1
-        if self.WAIT_MSG in self._driver.page_source():
+        if self.WAIT_MSG in self.page_source:
             raise BrowserError(f"Page failed to load within {self.MAX_WAIT} s")
 
     def find_element(self, by=None, value=None, bypass=True, wait=True):