diff --git a/modules/ImageContainer.py b/modules/ImageContainer.py
index b8f0e93..9061258 100644
--- a/modules/ImageContainer.py
+++ b/modules/ImageContainer.py
@@ -44,7 +44,7 @@ class ImageContainer:
 
     RETURN_TYPES = ("IMAGE",)
     FUNCTION = "image_container"
-    CATEGORY = "mask"
+    CATEGORY = "image"
 
     def image_container(self, width, height, red, green, blue, alpha):
         return (create_rgba_image(width, height, (red, green, blue, int(alpha * 255))).image_to_tensor().unsqueeze(0),)
@@ -93,7 +93,7 @@ class ImageContainerInheritanceAdd:
 
     RETURN_TYPES = ("IMAGE",)
     FUNCTION = "image_container_inheritance_add"
-    CATEGORY = "mask"
+    CATEGORY = "image"
 
     def image_container_inheritance_add(self, images, add_width, add_height, red, green, blue, alpha, method):
         width, height = images[0, :, :, 0].shape
@@ -156,7 +156,7 @@ class ImageContainerInheritanceScale:
 
     RETURN_TYPES = ("IMAGE",)
     FUNCTION = "image_container_inheritance_scale"
-    CATEGORY = "mask"
+    CATEGORY = "image"
 
     def image_container_inheritance_scale(self, images, scale_width, scale_height, red, green, blue, alpha, method):
         width, height = images[0, :, :, 0].shape
@@ -204,7 +204,7 @@ class ImageContainerInheritanceMax:
 
     RETURN_TYPES = ("IMAGE",)
     FUNCTION = "image_container_inheritance_max"
-    CATEGORY = "mask"
+    CATEGORY = "image"
 
     def image_container_inheritance_max(self, images_a, images_b, red, green, blue, alpha, method):
         img_a_height, img_a_width = images_a[0, :, :, 0].shape
@@ -265,7 +265,7 @@ class ImageContainerInheritanceSum:
 
     RETURN_TYPES = ("IMAGE",)
     FUNCTION = "image_container_inheritance_sum"
-    CATEGORY = "mask"
+    CATEGORY = "image"
 
     def image_container_inheritance_sum(self, images_a, images_b, red, green, blue, alpha, container_size_type, method):
         img_a_height, img_a_width = images_a[0, :, :, 0].shape