diff --git a/python/tests/utils/game_controller.py b/python/tests/utils/game_controller.py
index c1058a98..e3b3ef5d 100644
--- a/python/tests/utils/game_controller.py
+++ b/python/tests/utils/game_controller.py
@@ -192,11 +192,11 @@ def jump(self):
   def _step(self, actions, steps):
     try:
       return self._env.step(actions, steps)
-    except RuntimeError:
+    except RuntimeError as e:
       if self._env.is_running():
         raise
       else:
-        raise EpisodeFinishedError()
+        raise EpisodeFinishedError from e
 
   @property
   def position(self):