diff --git a/src/components/Editor/Runners/PyodideRunner/PyodideRunner.jsx b/src/components/Editor/Runners/PyodideRunner/PyodideRunner.jsx
index fa0e19ed4..6c74c1f35 100644
--- a/src/components/Editor/Runners/PyodideRunner/PyodideRunner.jsx
+++ b/src/components/Editor/Runners/PyodideRunner/PyodideRunner.jsx
@@ -210,11 +210,9 @@ const PyodideRunner = () => {
   };
 
   const getInputElement = () => {
-    const pageInput = document.getElementById("input");
-    const webComponentInput = document.querySelector("editor-wc")
+    return document.querySelector("editor-wc")
       ? document.querySelector("editor-wc").shadowRoot.getElementById("input")
-      : null;
-    return pageInput || webComponentInput;
+      : document.getElementById("input");
   };
 
   const getInputContent = async (element) => {