Update fallback selector

Shopify · Mar 15, 2024 · fd29e07 · fd29e07
1 parent d02da66
commit fd29e07
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/assets/media-gallery.js b/assets/media-gallery.js
@@ -29,11 +29,16 @@ if (!customElements.get('media-gallery')) {
       }
 
       setActiveMedia(mediaId) {
-        const activeMedia = this.elements.viewer.querySelector(`[data-media-id="${mediaId}"],[data-media-id]`);
+        const activeMedia =
+          this.elements.viewer.querySelector(`[data-media-id="${mediaId}"]`) ||
+          this.elements.viewer.querySelector('[data-media-id]');
+        if (!activeMedia) {
+          return;
+        }
         this.elements.viewer.querySelectorAll('[data-media-id]').forEach((element) => {
           element.classList.remove('is-active');
         });
-        activeMedia.classList.add('is-active');
+        activeMedia?.classList?.add('is-active');
 
         this.preventStickyHeader();
         window.setTimeout(() => {