diff --git a/launch.py b/launch.py
index 0a5fb78c3..32fbe0c8f 100644
--- a/launch.py
+++ b/launch.py
@@ -4,8 +4,8 @@
 
 print('[System ARGV] ' + str(sys.argv))
 
-# root = os.path.dirname(os.path.abspath(__file__))
-root = ("/content/drive/MyDrive/Colab Notebooks/fooocus")
+root = os.path.dirname(os.path.abspath(__file__))
+# root = ("/content/drive/MyDrive/Colab Notebooks/fooocus")
 sys.path.append(root)
 os.chdir(root)