diff --git a/Disco_Diffusion.ipynb b/Disco_Diffusion.ipynb
index ee4d7c8..5cfa846 100644
--- a/Disco_Diffusion.ipynb
+++ b/Disco_Diffusion.ipynb
@@ -1068,7 +1068,7 @@
         "      \n",
         "      # Inits if not video frames\n",
         "      if args.animation_mode != \"Video Input\":\n",
-        "        if args.init_image == '' or args.init_image.lower() == 'none':\n",
+        "        if args.init_image in ['','none', 'None', 'NONE']:\n",
         "          init_image = None\n",
         "        else:\n",
         "          init_image = args.init_image\n",