diff --git a/AlphaFold2.ipynb b/AlphaFold2.ipynb
index ff37ebdc..588f5319 100644
--- a/AlphaFold2.ipynb
+++ b/AlphaFold2.ipynb
@@ -262,8 +262,8 @@
         "#@markdown - set dpi for image resolution\n",
         "\n",
         "if save_to_google_drive:\n",
-        "  from pydrive.drive import GoogleDrive\n",
-        "  from pydrive.auth import GoogleAuth\n",
+        "  from pydrive2.drive import GoogleDrive\n",
+        "  from pydrive2.auth import GoogleAuth\n",
         "  from google.colab import auth\n",
         "  from oauth2client.client import GoogleCredentials\n",
         "  auth.authenticate_user()\n",
@@ -633,4 +633,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}