diff --git a/notebooks/bonus-unit1/bonus-unit1.ipynb b/notebooks/bonus-unit1/bonus-unit1.ipynb
index ccec1ac..52b0835 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 https://github.com/huggingface/ml-agents/"
+        "!git clone --depth 1 --branch hf-integration https://github.com/huggingface/ml-agents"
       ]
     },
     {