diff --git a/__init__.py b/__init__.py
index a499d4c..ff2d041 100644
--- a/__init__.py
+++ b/__init__.py
@@ -11,4 +11,12 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "CustomTileMerger": "🔧 Advanced Tile Merger",
 }
 
+# Extend ComfyUI Type Definitions
+try:
+    from comfy.graph_utils import GraphBuilder
+    # Register custom dict type check
+    GraphBuilder.add_node_type("TILE_CONFIG", lambda x: isinstance(x, dict) and "splits" in x)
+except ImportError:
+    pass
+
 __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]