diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index 3ea05ba4..2a394de4 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -3295,10 +3295,10 @@ function refreshGallery(newsearch = false) {
                 return
             }
             json.forEach(item => {
-                container.appendChild(galleryImage(item))
+                galleryImageContainer.appendChild(galleryImage(item))
             })
             // Wait for all images to be loaded
-            Promise.all(Array.from(container.querySelectorAll("img")).map(img => {
+            Promise.all(Array.from(galleryImageContainer.querySelectorAll("img")).map(img => {
                 if (img.complete)
                 {
                     return Promise.resolve(img.naturalHeight !== 0)