diff --git a/deformable_gym/envs/mujoco/grasp_env.py b/deformable_gym/envs/mujoco/grasp_env.py
index bcb2cef..d665991 100644
--- a/deformable_gym/envs/mujoco/grasp_env.py
+++ b/deformable_gym/envs/mujoco/grasp_env.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Any, Dict, Tuple
 
 import mujoco
@@ -14,14 +16,6 @@ class GraspEnv(BaseMJEnv):
     A custom MuJoCo environment for a grasping task, where a robot attempts to grasp an object.
     """
 
-    meta_data = {
-        "render_mode": [
-            "human",
-            "rgb_array",
-            "depth_array",
-        ],
-    }
-
     def __init__(
         self,
         robot_name: str,
@@ -30,7 +24,7 @@ def __init__(
         observable_object_pos: bool = True,
         control_type: str = "mocap",
         max_sim_time: float = 6,
-        render_mode: str | None = "human",
+        render_mode: str | None = None,
         mocap_cfg: Dict[str, str] | None = None,
         init_frame: str | None = None,
         default_cam_config: Dict[str, Any] | None = None,
diff --git a/deformable_gym/helpers/mj_utils.py b/deformable_gym/helpers/mj_utils.py
index f627339..da191b7 100644
--- a/deformable_gym/helpers/mj_utils.py
+++ b/deformable_gym/helpers/mj_utils.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Dict, List, Tuple
 
diff --git a/deformable_gym/robots/mj_robot.py b/deformable_gym/robots/mj_robot.py
index 117fe20..50f0089 100644
--- a/deformable_gym/robots/mj_robot.py
+++ b/deformable_gym/robots/mj_robot.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from abc import ABC
 from typing import List