diff --git a/src/primaite/session/environment.py b/src/primaite/session/environment.py
index 6c42c701..4d0544e9 100644
--- a/src/primaite/session/environment.py
+++ b/src/primaite/session/environment.py
@@ -83,7 +83,7 @@ class PrimaiteGymEnv(gymnasium.Env):
         with open(path, "w") as file:
             json.dump(data, file)
 
-    def reset(self, seed: Optional[int] = None) -> Tuple[ObsType, Dict[str, Any]]:
+    def reset(self, seed: Optional[int] = None, options: Optional[Dict] = None) -> Tuple[ObsType, Dict[str, Any]]:
         """Reset the environment."""
         _LOGGER.info(
             f"Resetting environment, episode {self.episode_counter}, "