diff --git a/deformable_gym/envs/mujoco/base_mjenv.py b/deformable_gym/envs/mujoco/base_mjenv.py
index 5582897..0a3caae 100644
--- a/deformable_gym/envs/mujoco/base_mjenv.py
+++ b/deformable_gym/envs/mujoco/base_mjenv.py
@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Any, Optional, Tuple
+from typing import Any, Dict, Optional, Tuple
 
 import gymnasium as gym
 import mujoco
@@ -88,7 +88,7 @@ def reset(
         *,
         seed: Optional[int] = None,
         options: Optional[dict] = None,
-    ) -> Tuple[NDArray[np.float64], dict[str, Any]]:
+    ) -> Tuple[NDArray[np.float64], Dict[str, Any]]:
         """
         Resets the environment to its initial state.