diff --git a/game/end_to_end_tests/test_level_selection.py b/game/end_to_end_tests/test_level_selection.py
index f82160104..adb3405e6 100644
--- a/game/end_to_end_tests/test_level_selection.py
+++ b/game/end_to_end_tests/test_level_selection.py
@@ -68,12 +68,12 @@ def test_redirect_to_levelless_episode(self):
 
         page = self.go_to_level("41", True)
 
-        self.wait_for_element_to_be_clickable((By.ID, "next_button"))
-        self.find_element(By.ID, "next_button").click()
+        page.wait_for_element_to_be_clickable((By.ID, "next_button"))
+        page.find_element(By.ID, "next_button").click()
 
         current_url = page.browser.current_url
         assert current_url == expected_url
 
-        episode_20_header = self.find_element(By.ID, "episode-20")
+        episode_20_header = page.find_element(By.ID, "episode-20")
         episode_20_expanded = episode_20_header.get_attribute("aria-expanded")
         assert episode_20_expanded == True