diff --git a/__init__.py b/__init__.py
index 15268c1..1b761cf 100644
--- a/__init__.py
+++ b/__init__.py
@@ -4,3 +4,8 @@ NODE_CLASS_MAPPINGS = {
     "WanMoeKSampler":WanMoeKSampler,
     "WanMoeKSamplerAdvanced":WanMoeKSamplerAdvanced
 }
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "WanMoeKSampler": "Wan MoE KSampler",
+    "WanMoeKSamplerAdvanced": "Wan MoE KSampler (Advanced)"
+}
\ No newline at end of file