update approach based on review's feedback

Shopify · May 6, 2024 · ef0cf34 · ef0cf34
1 parent 1d98d3c
commit ef0cf34
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/assets/global.js b/assets/global.js
@@ -1103,10 +1103,9 @@ class VariantSelects extends HTMLElement {
   }
 
   updateMedia(html) {
+    const sectionId = this.dataset.originalSection ? this.dataset.originalSection : this.dataset.section;
     const mediaGallerySource = document.querySelector(`[id^="MediaGallery-${this.dataset.section}"] ul`);
-    const mediaGalleryDestination = html.querySelector(
-      `[id^="MediaGallery-${this.dataset.section.replace('quickadd-', '')}"] ul`
-    );
+    const mediaGalleryDestination = html.querySelector(`[id^="MediaGallery-${sectionId}"] ul`);
 
     const refreshSourceData = () => {
       const mediaGallerySourceItems = Array.from(mediaGallerySource.querySelectorAll('li[data-media-id]'));