diff --git a/torch_training/render_agent_behavior.py b/torch_training/render_agent_behavior.py
index f98318e704d776d47937b50e7e452eab51355ee9..1d760c6ee60c62286aa5589ff358449daa5f8ed5 100644
--- a/torch_training/render_agent_behavior.py
+++ b/torch_training/render_agent_behavior.py
@@ -100,7 +100,7 @@ for trials in range(1, n_trials + 1):
 
     # Run episode
     for step in range(max_steps):
-        env_renderer.render_env(show=True, show_observations=False, show_predictions=True)
+        env_renderer.render_env(show=True, show_observations=False, show_predictions=False)
 
         if record_images:
             env_renderer.gl.saveImage("./Images/flatland_frame_{:04d}.bmp".format(frame_step))