diff --git a/static/app.js b/static/app.js
index 599edab..84411df 100644
--- a/static/app.js
+++ b/static/app.js
@@ -1353,9 +1353,15 @@ async function loadGallery() {
         galleryGrid.innerHTML = "";
         
         outputs.forEach(file => {
-            const basename = file.filename.replace("upscaled_", "");
-            const jobId = basename.split(".")[0];
-            const matchingJob = jobs.find(j => j.job_id === jobId);
+            const matchingJob = jobs.find(j => {
+                if (j.output_file) {
+                    const jobFilename = j.output_file.split("/").pop();
+                    if (jobFilename === file.filename) return true;
+                }
+                const basename = file.filename.replace("upscaled_", "");
+                const jobId = basename.split(".")[0];
+                return j.job_id === jobId;
+            });
             
             const card = document.createElement("div");
             card.className = "gallery-card";