diff --git a/__init__.py b/__init__.py
index 3bad042..744d3d5 100644
--- a/__init__.py
+++ b/__init__.py
@@ -84,6 +84,7 @@ class MZ_StylizePhotoControlNetApply:
         return {"required": required}
 
     RETURN_TYPES = ("StylizePhoto_CONTROL_NET",)
+    RETURN_NAMES = ("control_net",)
     FUNCTION = "load_controlnet"
     CATEGORY = CATEGORY_NAME