diff --git a/sumo_rl/environment/env.py b/sumo_rl/environment/env.py
index 38ced601..46fd1dbe 100755
--- a/sumo_rl/environment/env.py
+++ b/sumo_rl/environment/env.py
@@ -517,6 +517,7 @@ def __init__(self, **kwargs):
 
         self.seed()
         self.env = SumoEnvironment(**self._kwargs)
+        self.render_mode = self.env.render_mode
 
         self.agents = self.env.ts_ids
         self.possible_agents = self.env.ts_ids