diff --git a/urlchecker/core/urlproc.py b/urlchecker/core/urlproc.py index 125ad0b..6f1eec5 100644 --- a/urlchecker/core/urlproc.py +++ b/urlchecker/core/urlproc.py @@ -153,10 +153,11 @@ def get_driver(self, port: Optional[int] = None, timeout: Optional[int] = 5): try: from .webdriver import WebDriver - driver = WebDriver(port=port, timeout=timeout) + default_driver = WebDriver(port=port, timeout=timeout) - # Do a sanity check of the driver - driver.check("https://google.com") + # Do a sanity check of the default driver + default_driver.check("https://google.com") + driver = default_driver except: logger.warning( "Issue with driver, results will be improved if you have it! Please match your version from https://googlechromelabs.github.io/chrome-for-testing"