diff --git a/__init__.py b/__init__.py
index 7e43d1c..3b669c5 100644
--- a/__init__.py
+++ b/__init__.py
@@ -4,5 +4,7 @@
 @description: A custom node for ComfyUI to store and retrieve data dynamically.
 """
 
-from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS, NODE_CATEGORY_MAPPINGS
-__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'NODE_CATEGORY_MAPPINGS']
+WEB_DIRECTORY = "./js"
+
+from .bridge_node import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']