diff --git a/Unsloth_Studio_Colab.ipynb b/Unsloth_Studio_Colab.ipynb
index 4f4767aaac..bedaea85c0 100644
--- a/Unsloth_Studio_Colab.ipynb
+++ b/Unsloth_Studio_Colab.ipynb
@@ -15,9 +15,9 @@
     "import os\n",
     "github_token = os.environ.get('GITHUB_TOKEN', '')\n",
     "if github_token:\n",
-    "    !git clone https://{github_token}@github.com/unslothai/new-ui-prototype.git\n",
+    "    !git clone -b feature/colab-notebook https://{github_token}@github.com/unslothai/new-ui-prototype.git\n",
     "else:\n",
-    "    !git clone https://github.com/unslothai/new-ui-prototype.git\n",
+    "    !git clone -b feature/colab-notebook https://github.com/unslothai/new-ui-prototype.git\n",
     "%cd /content/new-ui-prototype\n",
     "\n",
     "# Run setup script\n",