diff --git a/units/en/unitbonus1/train.mdx b/units/en/unitbonus1/train.mdx
index 0ac738b..8e5c898 100644
--- a/units/en/unitbonus1/train.mdx
+++ b/units/en/unitbonus1/train.mdx
@@ -68,7 +68,7 @@ Before diving into the notebook, you need to:
 
 ```bash
 # Clone this specific repository (can take 3min)
-git clone https://github.com/huggingface/ml-agents/
+git clone --depth 1 --branch hf-integration https://github.com/huggingface/ml-agents
 ```
 
 ```bash