diff --git a/yolo_seg.py b/yolo_seg.py
index 58509c9..022fd44 100644
--- a/yolo_seg.py
+++ b/yolo_seg.py
@@ -70,7 +70,7 @@ class YoloSegNode:
         }
 
     RETURN_TYPES = ("IMAGE", "MASK",)
-    RETURN_NAMES = ("SEG_MASK", "SEG_IMAGE",)
+    RETURN_NAMES = ("SEG_IMAGE", "SEG_MASK",)
     FUNCTION = "seg"
     CATEGORY = "Jags_vector/yoloSEG"