diff --git a/random_size.py b/random_size.py
index 393d62d..b36b81f 100644
--- a/random_size.py
+++ b/random_size.py
@@ -8,9 +8,11 @@ def read_config(item):
 
 class RandomSize:
     CATEGORY = "utils"
-    @classmethod    
-    def INPUT_TYPES(s):
-        return { "required":{} }
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {},
+        }
     RETURN_TYPES = ("INT","INT")
     RETURN_NAMES = ("width","height")
     FUNCTION = "func"