diff --git a/torch_training/render_agent_behavior.py b/torch_training/render_agent_behavior.py
index 87acc2702445771fdd0226666310dd6d69568fc3..589d6109641c540c103f030e7eb139c35df25298 100644
--- a/torch_training/render_agent_behavior.py
+++ b/torch_training/render_agent_behavior.py
@@ -87,7 +87,7 @@ action_size = 5
 
 # We set the number of episodes we would like to train on
 if 'n_trials' not in locals():
-    n_trials = 6000
+    n_trials = 60000
 max_steps = int(3 * (env.height + env.width))
 eps = 1.
 eps_end = 0.005