diff --git a/units/en/unitbonus1/train.mdx b/units/en/unitbonus1/train.mdx
index 8e5c898..20629fe 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 --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
 ```
 
 ```bash