diff --git a/delete_model_passthrough.py b/delete_model_passthrough.py
index 138f9fd..6b64f92 100644
--- a/delete_model_passthrough.py
+++ b/delete_model_passthrough.py
@@ -16,12 +16,12 @@ class DeleteModelPassthrough:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "data": ("*",),        # any type (IMAGE, LATENT, CLIP, etc.)
-                "model": ("MODEL",),   # model to delete
-            }
+                "data": (any,),
+                "model": (any,),
+            },
         }
-
-    RETURN_TYPES = ("*",)
+ 
+    RETURN_TYPES = (any,)
     RETURN_NAMES = ("output",)
     FUNCTION = "run"
     CATEGORY = "Memory Management"