diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..ae741e5
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,3 @@
+from .SegMoE import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']