1 files changed, 3 insertions, 3 deletions
diff --git a/test/haketilo_test/conftest.py b/test/haketilo_test/conftest.py
index 9103ac8..834f026 100644
--- a/test/haketilo_test/conftest.py
+++ b/test/haketilo_test/conftest.py
@@ -46,7 +46,7 @@ def proxy():
 
 @pytest.fixture(scope="session")
 def _driver(proxy):
-    with firefox_safe_mode() as driver:
+    with firefox_safe_mode(proxy.server_port) as driver:
         yield driver
         driver.quit()
 
@@ -57,14 +57,14 @@ def close_all_but_one_window(driver):
     driver.switch_to.window(driver.window_handles[0])
 
 @pytest.fixture()
-def driver(_driver, request):
+def driver(proxy, _driver, request):
     nav_target = request.node.get_closest_marker('get_page')
     nav_target = nav_target.args[0] if nav_target else 'about:blank'
 
     second_driver = request.node.get_closest_marker('second_driver')
 
     if second_driver:
-        with firefox_safe_mode() as _driver:
+        with firefox_safe_mode(proxy.server_port) as _driver:
             _driver.get(nav_target)
             yield _driver
             _driver.quit()