diff --git a/bindsnet/environment/environment.py b/bindsnet/environment/environment.py
index aa893693..d10c4ed1 100644
--- a/bindsnet/environment/environment.py
+++ b/bindsnet/environment/environment.py
@@ -89,7 +89,7 @@ def __init__(
             2D inputs.
         """
         self.name = name
-        self.env = gym.make(env_id=name)  # , render_mode=render_mode)
+        self.env = gym.make(id=name, render_mode=render_mode)
         self.action_space = self.env.action_space
 
         self.encoder = encoder