diff --git a/unit2/unit2.ipynb b/unit2/unit2.ipynb
index 57deb18..e122820 100644
--- a/unit2/unit2.ipynb
+++ b/unit2/unit2.ipynb
@@ -1020,6 +1020,8 @@
         "                commit_message=\"Push Q-Learning agent to Hub\",\n",
         "                token= None\n",
         "                ):\n",
+        "  _, repo_name = repo_id.split(\"/\")\n",
+        "\n",
         "  eval_env = env\n",
         "  \n",
         "  # Step 1: Clone or create the repo\n",