diff --git a/__init__.py b/__init__.py
index 6e0a34d..e90ed4a 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,24 +1,6 @@
 from .node import *
 from .install import *
 
-NODE_CLASS_MAPPINGS = {
-    'SDVN SAMModelLoader': SAMModelLoader,
-    'SDVN DinoModelLoader': GroundingDinoModelLoader,
-    'SDVN DinoSAMSegment': GroundingDinoSAMSegment,
-    'SDVN InvertMask': InvertMask,
-    "SDVN IsMaskEmpty": IsMaskEmptyNode,
-    "SDVN SegmentAnything": SegmentAnything,
-}
-
-NODE_DISPLAY_NAME_MAPPINGS = {
-    '🧩 SAMModel Loader': 'SDVN SAMModelLoader',
-    '🧩 DinoModel Loader': 'SDVN DinoModelLoader',
-    '🧩 Dino SAMSegment': 'SDVN DinoSAMSegment',
-    '🧩 Invert Mask': 'SDVN InvertMask',
-    "🧩 Is Mask Empty": 'SDVN IsMaskEmpty',
-    "🧩 Segment Anything": 'SDVN SegmentAnything',
-}
-
-__all__ = ['NODE_CLASS_MAPPINGS']
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
 
 
diff --git a/node.py b/node.py
index 4be0c09..2acd6d9 100644
--- a/node.py
+++ b/node.py
@@ -427,3 +427,21 @@ class IsMaskEmptyNode:
 
     def main(self, mask):
         return (torch.all(mask == 0).int().item(), )
+
+NODE_CLASS_MAPPINGS = {
+    'SDVN SAMModelLoader': SAMModelLoader,
+    'SDVN DinoModelLoader': GroundingDinoModelLoader,
+    'SDVN DinoSAMSegment': GroundingDinoSAMSegment,
+    'SDVN InvertMask': InvertMask,
+    "SDVN IsMaskEmpty": IsMaskEmptyNode,
+    "SDVN SegmentAnything": SegmentAnything,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    'SDVN SAMModelLoader': '🧩 SAMModel Loader',
+    'SDVN DinoModelLoader': '🧩 DinoModel Loader',
+    'SDVN DinoSAMSegment': '🧩 Dino SAMSegment',
+    'SDVN InvertMask': '🧩 Invert Mask',
+    "SDVN IsMaskEmpty": '🧩 Is Mask Empty',
+    "SDVN SegmentAnything": '🧩 Segment Anything',
+}
\ No newline at end of file