diff --git a/__init__.py b/__init__.py
index 136d8b3..1bbf304 100644
--- a/__init__.py
+++ b/__init__.py
@@ -5,7 +5,7 @@
 @description: A custom node that pauses the flow while you choose which image or images to pass on to the rest of the workflow. Simplified and improved version of cg-image-picker.
 """
 
-from .image_filter import ImageFilter, TextImageFilter, MaskEditFilter
+from .image_filter import ImageFilter, TextImageFilter, MaskImageFilter
 
 VERSION = "0.1"
 WEB_DIRECTORY = "./js"
@@ -13,7 +13,7 @@ WEB_DIRECTORY = "./js"
 NODE_CLASS_MAPPINGS= {
     "Image Filter": ImageFilter,
     "Text Image Filter": TextImageFilter,
-    "Mask Edit Filter": MaskEditFilter
+    "Mask Image Filter": MaskImageFilter
 }
 
 __all__ = ["NODE_CLASS_MAPPINGS", "WEB_DIRECTORY"]
diff --git a/image_filter.py b/image_filter.py
index 574c3e6..9f2905d 100644
--- a/image_filter.py
+++ b/image_filter.py
@@ -111,7 +111,7 @@ class TextImageFilter(PreviewImage):
 
         return (image, response)
     
-class MaskEditFilter(PreviewImage, LoadImage):
+class MaskImageFilter(PreviewImage, LoadImage):
     RETURN_TYPES = ("IMAGE","MASK")
     RETURN_NAMES = ("image","mask")
     FUNCTION = "func"