fix frame detection in the example

StampyAI · Sep 15, 2023 · 7dfb227 · 7dfb227
1 parent 63acd09
commit 7dfb227
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/public/embed-example.html b/public/embed-example.html
@@ -27,7 +27,7 @@ <h3>Both search and initial questions, opening inline, expanding iframe height,
     const iframes = [...document.querySelectorAll('iframe')]
     window.addEventListener('message', function (event) {
       if (event.data.type === 'aisafety.info__height') {
-        const iframe = iframes.find((iframe) => iframe.src === event.source.location.href)
+        const iframe = event.source.frameElement
         if (iframe) {
           iframe.style.height = event.data.height + 'px'
         }