diff --git a/__init__.py b/__init__.py
index 4915092..0e89471 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-from nodes.Prompts import DOC_SaveImageAndAddToHistory
+from nodes.Prompts.DOC_SaveImageAndAddToHistory import DOC_SaveImageAndAddToHistory
 
 
 # A dictionary that contains all nodes you want to export with their names