diff --git a/__init__.py b/__init__.py
index 71d70e0..8aad30c 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-from stability_api import StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityCore, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3
+from .stability_api import StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityCore, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3
 
 NODE_CLASS_MAPPINGS = {
     "Stability Creative Upscale": StabilityCreativeUpscale,