diff --git a/scripts/daam_script.py b/scripts/daam_script.py
index bc5269b..5dfefbe 100644
--- a/scripts/daam_script.py
+++ b/scripts/daam_script.py
@@ -288,7 +288,9 @@ def before_image_saved(self, params : script_callbacks.ImageSaveParams):
                             img : Image.Image = utils.image_overlay_heat_map(params.image, heat_map_img, alpha=self.alpha, caption=caption, image_scale=self.heatmap_image_scale)
 
                             fullfn_without_extension, extension = os.path.splitext(params.filename) 
-                            full_filename = fullfn_without_extension + "_" + attention +  ("_" + self.attn_captions[i] if self.attn_captions[i] else "") + extension
+                            if not os.path.exists(fullfn_without_extension):
+                                os.makedirs(fullfn_without_extension)
+                            full_filename = fullfn_without_extension + "/" + attention + ("_" + self.attn_captions[i] if self.attn_captions[i] else "") + extension
                             
                             if self.use_grid:
                                 heatmap_images.append(img)