diff --git a/notebooks/unit2/unit2.ipynb b/notebooks/unit2/unit2.ipynb
index 7466aac..c7477e6 100644
--- a/notebooks/unit2/unit2.ipynb
+++ b/notebooks/unit2/unit2.ipynb
@@ -926,6 +926,7 @@
         "  \"\"\"\n",
         "  Evaluate the agent for ``n_eval_episodes`` episodes and returns average reward and std of reward.\n",
         "  :param env: The evaluation environment\n",
+        "  :param max_steps: Maximum number of steps per episode\n",
         "  :param n_eval_episodes: Number of episode to evaluate the agent\n",
         "  :param Q: The Q-table\n",
         "  :param seed: The evaluation seed array (for taxi-v3)\n",