diff --git a/notebooks/bonus-unit1/bonus-unit1.ipynb b/notebooks/bonus-unit1/bonus-unit1.ipynb
index fb128e4..5f87f23 100644
--- a/notebooks/bonus-unit1/bonus-unit1.ipynb
+++ b/notebooks/bonus-unit1/bonus-unit1.ipynb
@@ -174,7 +174,7 @@
       "source": [
         "%%capture\n",
         "# Clone this specific repository (can take 3min)\n",
-        "!git clone --depth 1 --branch hf-integration https://github.com/huggingface/ml-agents"
+        "!git clone --depth 1 --branch hf-integration-save https://github.com/huggingface/ml-agents"
       ]
     },
     {
@@ -582,4 +582,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}