diff --git a/utils/env_utils.py b/utils/env_utils.py
index eb57452..78ef69b 100644
--- a/utils/env_utils.py
+++ b/utils/env_utils.py
@@ -134,9 +134,9 @@ def get_images(self) -> Sequence[np.ndarray]:
 
     def render(self, mode: str = "human"):
         if self.num_envs == 1:
-            return self.envs[0].render(mode=mode)
+            return self.envs[0].render()
         else:
-            return super().render(mode=mode)
+            return super().render()
 
     def get_attr(self, attr_name, indices=None):
         """Return attribute from vectorized environment (see base class)."""