diff --git a/a2c_sb3_cartpole.py b/a2c_sb3_cartpole.py
index 724585316fa14b9aac3a639655727eba35ddd4eb..7c94adcbbdf8cb5b3af5c7def98c89e4e7b4bd90 100644
--- a/a2c_sb3_cartpole.py
+++ b/a2c_sb3_cartpole.py
@@ -20,7 +20,7 @@ run = wandb.init(
     save_code=True,
 )
 
-env = gym.make("CartPole-v1")
+env = gym.make("CartPole-v1", render_mode="rgb_array")
 
 model = A2C("MlpPolicy", env, verbose=1, tensorboard_log=f"runs/{run.id}")
 #model = A2C("MlpPolicy", env, )