diff --git a/game/end_to_end_tests/test_level_win.py b/game/end_to_end_tests/test_level_win.py
index bc2d14cdd..380755fef 100644
--- a/game/end_to_end_tests/test_level_win.py
+++ b/game/end_to_end_tests/test_level_win.py
@@ -2,6 +2,7 @@
 from game.models import Level
 
 from game.character import get_character
+from game.theme import get_theme
 
 class TestLevelWin(BaseGameTest):
     custom_level = None
@@ -41,7 +42,7 @@ def setUpClass(cls):
             origin='{"coordinate":[3,5],"direction":"S"}',
             path='[{"coordinate":[3,5],"connectedNodes":[1]},{"coordinate":[3,4],' '"connectedNodes":[0]}]',
             pythonEnabled=False,
-            theme=grass,
+            theme=get_theme(name="grass"),
             threads=1,
             traffic_lights="[]",
         )