diff --git a/docs/assets/plugins/code-block/code-block.js b/docs/assets/plugins/code-block/code-block.js
index dad9e3be8..b6ed38d5c 100644
--- a/docs/assets/plugins/code-block/code-block.js
+++ b/docs/assets/plugins/code-block/code-block.js
@@ -7,13 +7,14 @@
 
   function runScript(script) {
     const newScript = document.createElement('script');
-    newScript.appendChild(
-      document.createTextNode(`
-      (() => {
-        ${script.innerHTML}
-      })();
-    `)
-    );
+
+    if (script.type === 'module') {
+      newScript.type = 'module';
+      newScript.textContent = script.innerHTML;
+    } else {
+      newScript.appendChild(document.createTextNode(`(() => { ${script.innerHTML} })();`));
+    }
+
     script.parentNode.replaceChild(newScript, script);
   }