diff --git a/src/st_keyup/frontend/main.js b/src/st_keyup/frontend/main.js
index eadc249..37ecb8b 100644
--- a/src/st_keyup/frontend/main.js
+++ b/src/st_keyup/frontend/main.js
@@ -90,11 +90,12 @@ function onRender(event) {
       input.onkeyup = onKeyUp
     }
 
+    // Render with the correct height
+    Streamlit.setFrameHeight(73)
+
     window.rendered = true
   }
 }
 
 Streamlit.events.addEventListener(Streamlit.RENDER_EVENT, onRender)
 Streamlit.setComponentReady()
-// Render with the correct height
-Streamlit.setFrameHeight(73)
\ No newline at end of file