diff --git a/notebooks/comfyui_colab.ipynb b/notebooks/comfyui_colab.ipynb
index 42275a3e6..5108ec830 100644
--- a/notebooks/comfyui_colab.ipynb
+++ b/notebooks/comfyui_colab.ipynb
@@ -30,13 +30,13 @@
         "OPTIONS['UPDATE_COMFY_UI'] = UPDATE_COMFY_UI\n",
         "\n",
         "if OPTIONS['USE_GOOGLE_DRIVE']:\n",
+        "    !echo \"Mounting Google Drive...\"\n",
+        "    %cd /\n",
+        "    \n",
         "    from google.colab import drive\n",
         "    drive.mount('/content/drive')\n",
         "\n",
-        "    global WORKSPACE\n",
-        "    WORKSPACE = str(\n",
-        "        Path('drive', 'MyDrive', WORKSPACE).resolve()\n",
-        "    )\n",
+        "    WORKSPACE = \"/content/drive/MyDrive/ComfyUI\"\n",
         "    %cd /content/drive/MyDrive\n",
         "\n",
         "![ ! -d $WORKSPACE ] && echo -= Initial setup ComfyUI =- && git clone https://github.com/comfyanonymous/ComfyUI\n",
@@ -46,6 +46,7 @@
         "  !echo -= Updating ComfyUI =-\n",
         "  !git pull\n",
         "\n",
+        "!echo -= Install dependencies =-\n",
         "!pip -q install xformers -r requirements.txt"
       ]
     },