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