diff --git a/javascript/imageviewer.js b/javascript/imageviewer.js
index e4dae91bc..5e657a51d 100644
--- a/javascript/imageviewer.js
+++ b/javascript/imageviewer.js
@@ -179,7 +179,7 @@ function modalTileImageToggle(event) {
 }
 
 onAfterUiUpdate(function() {
-    var fullImg_preview = gradioApp().querySelectorAll('.gradio-gallery > div > img');
+    var fullImg_preview = gradioApp().querySelectorAll('.gradio-gallery > button > button > img');
     if (fullImg_preview != null) {
         fullImg_preview.forEach(setupImageForLightbox);
     }