diff --git a/webpages/bufftools_pages/ai_image_editor.py b/webpages/bufftools_pages/ai_image_editor.py
index 8991eacd7..c57989644 100644
--- a/webpages/bufftools_pages/ai_image_editor.py
+++ b/webpages/bufftools_pages/ai_image_editor.py
@@ -75,14 +75,14 @@ def ai_image_editor():
                         st.info(text_response)
                     if image_response:
                         st.image(image_response, width=300)
-                    # Convert PIL image to bytes for download
-                    img_byte_arr = BytesIO()
-                    image_response.save(img_byte_arr, format="JPEG")
-                    st.download_button(
-                        label="Download Edited Image",
-                        data=img_byte_arr.getvalue(),
-                        file_name="edited_image.jpeg",
-                        mime="image/jpeg"
-                    )
+                        # Convert PIL image to bytes for download
+                        img_byte_arr = BytesIO()
+                        image_response.save(img_byte_arr, format="JPEG")
+                        st.download_button(
+                            label="Download Edited Image",
+                            data=img_byte_arr.getvalue(),
+                            file_name="edited_image.jpeg",
+                            mime="image/jpeg"
+                        )