diff --git a/scripts/intersection_social_dqn.ipynb b/scripts/intersection_social_dqn.ipynb
index 9cef8b0d4..e29f32acb 100644
--- a/scripts/intersection_social_dqn.ipynb
+++ b/scripts/intersection_social_dqn.ipynb
@@ -147,8 +147,8 @@
         "env = load_environment(env_config)\n",
         "env.config[\"offscreen_rendering\"] = True\n",
         "agent = load_agent(agent_config, env)\n",
-        "evaluation = Evaluation(env, agent, num_episodes=1)\n",
-        "evaluation.train()\n",
+        "evaluation = Evaluation(env, agent, num_episodes=20, training = False, recover = True)\n",
+        "evaluation.test()\n",
         "show_videos(evaluation.run_directory)"
       ]
     }