diff --git a/generate_equirect.py b/generate_equirect.py
index 85a048e..23920fd 100644
--- a/generate_equirect.py
+++ b/generate_equirect.py
@@ -12,7 +12,6 @@ import json
 import math
 import os
 import re
-import shutil
 import tempfile
 import torch
 from PIL import Image, ImageDraw
@@ -555,7 +554,9 @@ def generate(
             except Exception as e:  # noqa: BLE001
                 print(f"Upscaling failed ({upscale}); keeping seamless image: {e}")
 
-        shutil.move(final_path, output_path)
+        with Image.open(final_path) as final_img:
+            final_img.load()
+            save_png_with_prompt(final_img, output_path, prompt)
         try:
             with Image.open(output_path) as _im:
                 print(f"→ Final image written to {output_path} [{_im.size[0]}x{_im.size[1]}]")