diff --git a/game/end_to_end_tests/test_level_selection.py b/game/end_to_end_tests/test_level_selection.py
index 4655388fe..4757f12b6 100644
--- a/game/end_to_end_tests/test_level_selection.py
+++ b/game/end_to_end_tests/test_level_selection.py
@@ -69,7 +69,7 @@ def test_redirect_to_levelless_episode(self):
         expected_url = levels_page.browser.current_url
 
         page = self.go_to_level("41", True)
-        assert WebDriverWait(self.browser, 10).until(
+        assert WebDriverWait(self.selenium, 10).until(
             EC.element_to_be_clickable((By.ID, "next_button"))
         )
         page.browser.find_element(By.ID, "next_button").click()