diff --git a/notebooks/unit2/unit2.ipynb b/notebooks/unit2/unit2.ipynb
index a10b711..ea808f6 100644
--- a/notebooks/unit2/unit2.ipynb
+++ b/notebooks/unit2/unit2.ipynb
@@ -771,7 +771,7 @@
       "outputs": [],
       "source": [
         "def train(n_training_episodes, min_epsilon, max_epsilon, decay_rate, env, max_steps, Qtable):\n",
-        "  for episode in tqdm(range(n_training_episodes))\n",
+        "  for episode in tqdm(range(n_training_episodes)):\n",
         "    # Reduce epsilon (because we need less and less exploration)\n",
         "    epsilon = min_epsilon + (max_epsilon - min_epsilon)*np.exp(-decay_rate*episode)\n",
         "    # Reset the environment\n",