diff --git a/__init__.py b/__init__.py
index 27dadc6..a7145d4 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1 +1,2 @@
 from .cut_png_node import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']