diff --git a/client/coral-embed/src/index.js b/client/coral-embed/src/index.js
index d3a6a4ee5..8d8855aa7 100644
--- a/client/coral-embed/src/index.js
+++ b/client/coral-embed/src/index.js
@@ -36,7 +36,7 @@ function configurePymParent(pymParent, asset_url) {
   // Resize parent iframe height when child height changes
   pymParent.onMessage('height', function(height) {
     if (height !== cachedHeight) {
-      pymParent.el.querySelector('iframe').height = `${height  }px`;
+      pymParent.el.firstChild.style.height = `${height}px`;
       cachedHeight = height;
     }
   });