diff --git a/Disco_Diffusion.ipynb b/Disco_Diffusion.ipynb
index 48814e2..b56e04d 100644
--- a/Disco_Diffusion.ipynb
+++ b/Disco_Diffusion.ipynb
@@ -472,7 +472,7 @@
         "  subprocess.run(['apt', 'install', 'imagemagick'], stdout=subprocess.PIPE).stdout.decode('utf-8')\n",
         "\n",
         "try:\n",
-        "  import CLIP\n",
+        "  from CLIP import clip\n",
         "except:\n",
         "  if os.path.exists(\"CLIP\") is not True:\n",
         "    gitclone(\"https://github.com/openai/CLIP\")\n",
@@ -542,7 +542,7 @@
         "import torchvision.transforms as T\n",
         "import torchvision.transforms.functional as TF\n",
         "from tqdm.notebook import tqdm\n",
-        "import CLIP\n",
+        "from CLIP import clip\n",
         "from resize_right import resize\n",
         "from guided_diffusion.script_util import create_model_and_diffusion, model_and_diffusion_defaults\n",
         "from datetime import datetime\n",
diff --git a/disco.py b/disco.py
index d25daed..92d712c 100644
--- a/disco.py
+++ b/disco.py
@@ -448,7 +448,7 @@ if is_colab:
   subprocess.run(['apt', 'install', 'imagemagick'], stdout=subprocess.PIPE).stdout.decode('utf-8')
 
 try:
-  import CLIP
+  from CLIP import clip
 except:
   if os.path.exists("CLIP") is not True:
     gitclone("https://github.com/openai/CLIP")
@@ -518,7 +518,7 @@ from torch.nn import functional as F
 import torchvision.transforms as T
 import torchvision.transforms.functional as TF
 from tqdm.notebook import tqdm
-import CLIP
+from CLIP import clip
 from resize_right import resize
 from guided_diffusion.script_util import create_model_and_diffusion, model_and_diffusion_defaults
 from datetime import datetime