diff --git a/__init__.py b/__init__.py
index 0fd7cd3..ed1771a 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-from nodes.image import SaveImagePath
+from .nodes.image import SaveImagePath
 from .nodes.heygem import HeyGemF2F, HeyGemF2FFromFile
 from .nodes.s3 import S3Download, S3Upload, S3UploadURL
 from .nodes.text import *