diff --git a/__init__.py b/__init__.py
index 8037025..4915092 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-from nodes.prompts import DOC_SaveImageAndAddToHistory
+from nodes.Prompts import DOC_SaveImageAndAddToHistory
 
 
 # A dictionary that contains all nodes you want to export with their names
diff --git a/nodes/prompts/DOC_SaveImageAndAddToHistory.py b/nodes/prompts/DOC_SaveImageAndAddToHistory.py
index 3245674..fd65f45 100644
--- a/nodes/prompts/DOC_SaveImageAndAddToHistory.py
+++ b/nodes/prompts/DOC_SaveImageAndAddToHistory.py
@@ -47,7 +47,7 @@ class DOC_SaveImageAndAddToHistory:
 
     OUTPUT_NODE = True
 
-    CATEGORY = "image"
+    CATEGORY = "DOC"
     DESCRIPTION = "Saves the input images to your ComfyUI output directory and store prompt-image mapping in a global history file."
 
     def save_images(self, images, filename_prefix="ComfyUI", llm_prompt=None, final_prompt=None, steps=None, cfg=None, prompt=None, extra_pnginfo=None):