diff --git a/classifiers.py b/classifiers.py
index 91a504e..8b61ec8 100644
--- a/classifiers.py
+++ b/classifiers.py
@@ -43,8 +43,8 @@ class YOLOClassify:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "yolo_classifier_model": "YOLO_CLASSIFIER_MODEL",
-                "image_to_classify": "IMAGE"
+                "yolo_classifier_model": ("YOLO_CLASSIFIER_MODEL",),
+                "image_to_classify": ("IMAGE",)
             }
         }