diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..71d70e0
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,11 @@
+from stability_api import StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityCore, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3
+
+NODE_CLASS_MAPPINGS = {
+    "Stability Creative Upscale": StabilityCreativeUpscale,
+    "Stability Remove Background": StabilityRemoveBackground,
+    "Stability Inpainting": StabilityInpainting,
+    "Stability Image Core": StabilityCore,
+    "Stability Search and Replace": StabilitySearchAndReplace,
+    "Stability Outpainting": StabilityOutpainting,
+    "Stability SD3": StabilitySD3,
+}
diff --git a/stability_api.py b/stability_api.py
index b60e15b..e3c447f 100644
--- a/stability_api.py
+++ b/stability_api.py
@@ -258,14 +258,3 @@ class StabilityOutpainting(StabilityBase):
             "api_key override": ("STRING", {"multiline": False}),
         },
     }
-
-
-NODE_CLASS_MAPPINGS = {
-    "Stability Creative Upscale": StabilityCreativeUpscale,
-    "Stability Remove Background": StabilityRemoveBackground,
-    "Stability Inpainting": StabilityInpainting,
-    "Stability Image Core": StabilityCore,
-    "Stability Search and Replace": StabilitySearchAndReplace,
-    "Stability Outpainting": StabilityOutpainting,
-    "Stability SD3": StabilitySD3,
-}