diff --git a/modules/bgremoval.py b/modules/bgremoval.py
index 48dfd0c..15bec2e 100644
--- a/modules/bgremoval.py
+++ b/modules/bgremoval.py
@@ -81,7 +81,7 @@ class bgremoval:
     DESCRIPTION = "Remove backgrounds from images effortlessly using Runware's low-cost image editing Inference."
     FUNCTION = "rembg"
     RETURN_TYPES = ("STRING",)
-    RETURN_NAMES = ("Image",)
+    RETURN_NAMES = ("IMAGE",)
     CATEGORY = "Runware"
 
     def rembg(self, **kwargs):
diff --git a/modules/photoMaker.py b/modules/photoMaker.py
index ea5a116..c4afef8 100644
--- a/modules/photoMaker.py
+++ b/modules/photoMaker.py
@@ -113,7 +113,7 @@ class photoMaker:
     DESCRIPTION = "Transform and style images using PhotoMaker's advanced personalization technology. Create consistent, high-quality image variations with precise subject fidelity and style control."
     FUNCTION = "photoMaker"
     RETURN_TYPES = ("STRING",)
-    RETURN_NAMES = ("Image",)
+    RETURN_NAMES = ("IMAGE",)
     CATEGORY = "Runware"
 
     @classmethod