diff --git a/kikotools/__init__.py b/kikotools/__init__.py
index 785e4d8..08341e8 100644
--- a/kikotools/__init__.py
+++ b/kikotools/__init__.py
@@ -14,7 +14,7 @@ from .tools.image_scale_down_by import ImageScaleDownByNode
 from .tools.gemini_prompt import GeminiPromptNode
 from .tools.display_any import DisplayAnyNode
 from .tools.display_text import DisplayTextNode
-from .tools.xyz_grid import XYZPlotControllerAdvanced, ImageGridCombiner
+from .tools.xyz_grid import XYZPlotController, ImageGridCombiner
 
 # ComfyUI node registration mappings
 NODE_CLASS_MAPPINGS = {
@@ -30,7 +30,7 @@ NODE_CLASS_MAPPINGS = {
     "GeminiPrompt": GeminiPromptNode,
     "DisplayAny": DisplayAnyNode,
     "DisplayText": DisplayTextNode,
-    "XYZPlotController": XYZPlotControllerAdvanced,
+    "XYZPlotController": XYZPlotController,
     "ImageGridCombiner": ImageGridCombiner,
 }
 
diff --git a/kikotools/tools/xyz_grid/__init__.py b/kikotools/tools/xyz_grid/__init__.py
index 2a8807d..cc2c10b 100644
--- a/kikotools/tools/xyz_grid/__init__.py
+++ b/kikotools/tools/xyz_grid/__init__.py
@@ -13,4 +13,4 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "ImageGridCombiner": "Image Grid Combiner",
 }
 
-__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "XYZPlotController", "ImageGridCombiner"]
\ No newline at end of file
+__all__ = ["XYZPlotController", "ImageGridCombiner"]
\ No newline at end of file