diff --git a/birefnetNode.py b/birefnetNode.py
index 5199204..5d3b63a 100644
--- a/birefnetNode.py
+++ b/birefnetNode.py
@@ -102,7 +102,7 @@ def INPUT_TYPES(cls):
             }
         }
 
-    RETURN_TYPES = ("BiRefNetMODEL",)
+    RETURN_TYPES = ("BIREFNET",)
     RETURN_NAMES = ("model",)
     FUNCTION = "load_model"
     CATEGORY = "image/BiRefNet"
@@ -141,7 +141,7 @@ def INPUT_TYPES(cls):
             }
         }
 
-    RETURN_TYPES = ("BiRefNetMODEL",)
+    RETURN_TYPES = ("BIREFNET",)
     RETURN_NAMES = ("model",)
     FUNCTION = "load_model"
     CATEGORY = "rembg/BiRefNet"
@@ -179,7 +179,7 @@ class GetMaskByBiRefNet:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "model": ("BiRefNetMODEL",),
+                "model": ("BIREFNET",),
                 "images": ("IMAGE",),
                 "width": ("INT",
                           {
@@ -309,7 +309,7 @@ class RembgByBiRefNetAdvanced(GetMaskByBiRefNet, BlurFusionForegroundEstimation)
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "model": ("BiRefNetMODEL",),
+                "model": ("BIREFNET",),
                 "images": ("IMAGE",),
                 "width": ("INT",
                           {
@@ -358,7 +358,7 @@ class RembgByBiRefNet(RembgByBiRefNetAdvanced):
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "model": ("BiRefNetMODEL",),
+                "model": ("BIREFNET",),
                 "images": ("IMAGE",),
             }
         }