diff --git a/layered_diffusion.py b/layered_diffusion.py
index 5ffc79b..ce5fac7 100644
--- a/layered_diffusion.py
+++ b/layered_diffusion.py
@@ -164,6 +164,7 @@ def INPUT_TYPES(s):
 
     MAX_FRAMES = 3
     RETURN_TYPES = ("IMAGE",) * MAX_FRAMES
+    RETURN_NAMES = tuple(f"image{i + 1}" for i in range(MAX_FRAMES))
 
     def decode(
         self,
@@ -441,6 +442,7 @@ def INPUT_TYPES(s):
         }
 
     RETURN_TYPES = ("MODEL", "CONDITIONING", "CONDITIONING")
+    RETURN_NAMES = ("MODEL", "positive", "negative")
     FUNCTION = "apply_layered_diffusion"
     CATEGORY = "layer_diffuse"
     MODELS = (
@@ -566,6 +568,7 @@ def INPUT_TYPES(s):
         }
 
     RETURN_TYPES = ("MODEL", "CONDITIONING", "CONDITIONING")
+    RETURN_NAMES = ("MODEL", "positive", "negative")
     FUNCTION = "apply_layered_diffusion"
     CATEGORY = "layer_diffuse"
     MODELS = (