diff --git a/Reinforcement (Q-)Learning with PyTorch.ipynb b/Reinforcement (Q-)Learning with PyTorch.ipynb
index 54dbc5d2146..6383b1fc8a7 100644
--- a/Reinforcement (Q-)Learning with PyTorch.ipynb	
+++ b/Reinforcement (Q-)Learning with PyTorch.ipynb	
@@ -232,7 +232,7 @@
     "    # Resize, and add a batch dimension (BCHW)\n",
     "    return resize(screen).unsqueeze(0)\n",
     "\n",
-    "\n",
+    "env.reset()\n",
     "plt.imshow(get_screen().squeeze(0).permute(1, 2, 0).numpy(), interpolation='none')\n",
     "plt.show()"
    ]