diff --git a/notebooks/ComfyBox_Colab.ipynb b/notebooks/ComfyBox_Colab.ipynb
index 8f0b06e..338b1f2 100644
--- a/notebooks/ComfyBox_Colab.ipynb
+++ b/notebooks/ComfyBox_Colab.ipynb
@@ -5059,14 +5059,15 @@
         "\n",
         "![ ! -d $WORKSPACE ] && echo -= Initial setup ComfyUI =- && git clone https://github.com/comfyanonymous/ComfyUI\n",
         "%cd $WORKSPACE\n",
-        "![ ! -d \"ComfyBox\" ] && echo -= Initial setup ComfyBox =- && git clone --recursive https://github.com/space-nuko/ComfyBox -b subgraph-templates3 && cd ComfyBox && pnpm install && pnpm prebuild && pnpm build && cd ..\n",
+        "![ ! -d \"ComfyBox\" ] && echo -= Initial setup ComfyBox =- && git clone --recursive https://github.com/space-nuko/ComfyBox && cd ComfyBox && pnpm install && pnpm prebuild && pnpm build && cd ..\n",
         "\n",
         "if OPTIONS['UPDATE_COMFY_UI']:\n",
         "  !echo -= Updating ComfyUI =-\n",
         "  !git pull\n",
         "  !echo -= Updating ComfyBox =-\n",
         "  %cd ComfyBox\n",
-        "  !git pull\n",
+        "  !git fetch -a\n",
+        "  !git reset --hard origin/master\n",
         "  !pnpm install\n",
         "  !pnpm prebuild\n",
         "  !pnpm build\n",
@@ -5585,4 +5586,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}