diff --git a/cup.py b/cup.py
index f080336..1df0346 100644
--- a/cup.py
+++ b/cup.py
@@ -20,7 +20,7 @@ import logging as logger
 from comfy.cli_args import args
 try:
     from comfy.comfy_types import IO
-except ImportError:
+except (ImportError, ModuleNotFoundError):
     class IO:
         STRING = "STRING"
 try: