diff --git a/units/en/unit5/hands-on.mdx b/units/en/unit5/hands-on.mdx
index a5c867e..b78a25c 100644
--- a/units/en/unit5/hands-on.mdx
+++ b/units/en/unit5/hands-on.mdx
@@ -85,7 +85,7 @@ Before diving into the notebook, you need to:
 ```python
 %%capture
 # Clone the repository
-!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
 ```
 
 ```python