diff --git a/Lab2/lab2.ipynb b/Lab2/lab2.ipynb
index 7220745..4445217 100644
--- a/Lab2/lab2.ipynb
+++ b/Lab2/lab2.ipynb
@@ -1459,7 +1459,7 @@
     "            else:\n",
     "                action = sess.run(mainQN.predict, \n",
     "                                  feed_dict={mainQN.observation_input:[observation]})[0]\n",
-    "            if not train_model and np.random.rand(1) < 0.1:\n",
+    "            if not train_model and np.random.rand(1) < endE:\n",
     "                action = np.random.randint(0,action_space_size)\n",
     "            observations_1, reward, done = env.step(action)\n",
     "                                      \n",