diff --git a/src/captcha.ipynb b/src/captcha.ipynb
index 51f3c498cd3..7d6d774713e 100644
--- a/src/captcha.ipynb
+++ b/src/captcha.ipynb
@@ -143,10 +143,12 @@
         "    drive.mount(str(GoogleDriveManager._googleDriveFolder))\n",
         "\n",
         "  @staticmethod\n",
-        "  def uploadFolderToGoogleDrive(folder, zippedFolderName):\n",
-        "    !zip -r {zippedFolderName}.zip {folder}/\n",
-        "    !cp {zippedFolderName}.zip {GoogleDriveManager._baseFolder}\n",
-        "\n",
+        "  def uploadFolderToGoogleDrive(folder):\n",
+        "    basename = !basename {folder}\n",
+        "    basename = basename[0]\n",
+        "    !cd {folder}/..; zip -r {basename}.zip {basename}/\n",
+        "    !cd {folder}/..; cp {basename}.zip {GoogleDriveManager._baseFolder}\n",
+        "    \n",
         "  @staticmethod\n",
         "  def downloadFolderFromGoogleDrive(folder):\n",
         "    !cp {GoogleDriveManager._baseFolder}/{folder}.zip .\n",
@@ -320,7 +322,7 @@
         "    modelFilepath = f'{srcPath}/captcha/{model.name}'\n",
         "    modelDAO.saveModel(model, modelFilepath)\n",
         "    if inColab:\n",
-        "        GoogleDriveManager.uploadFolderToGoogleDrive(modelFilepath, model.name)"
+        "        GoogleDriveManager.uploadFolderToGoogleDrive(modelFilepath)"
       ]
     },
     {