diff --git a/cli/src/playground.html b/cli/src/playground.html
index b6955a97..018c4ced 100644
--- a/cli/src/playground.html
+++ b/cli/src/playground.html
@@ -90,10 +90,11 @@
   <script src="https://cdnjs.cloudflare.com/ajax/libs/clusterize.js/0.19.0/clusterize.min.js"></script>
 
   <script>LANGUAGE_BASE_URL = "";</script>
-  <script type="module" src=playground.js></script>
+  <script type="module" src="playground.js"></script>
   <script type="module">
-    import {initializePlayground} from './playground.js';
-    setTimeout(() => initializePlayground({local: true}), 1)
+    import * as TreeSitter from './tree-sitter.js';
+    window.TreeSitter = TreeSitter;
+    setTimeout(() => window.initializePlayground({local: true}), 1)
   </script>
 
   <style>
diff --git a/docs/src/7-playground.md b/docs/src/7-playground.md
index 201aac9c..90f97685 100644
--- a/docs/src/7-playground.md
+++ b/docs/src/7-playground.md
@@ -101,11 +101,10 @@ you must use at least one capture, like <code>(node_name) @capture-name</code></
 <script src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/6.65.7/codemirror.min.js"></script>
 
 <script>LANGUAGE_BASE_URL = "https://tree-sitter.github.io";</script>
-<script src="https://tree-sitter.github.io/tree-sitter.js"></script>
+<script type="module">
+import * as TreeSitter from 'https://tree-sitter.github.io/tree-sitter.js';
+window.TreeSitter = TreeSitter;
+setTimeout(() => window.initializePlayground({local: false}), 1);
+</script>
 
 <script src="https://cdnjs.cloudflare.com/ajax/libs/clusterize.js/0.19.0/clusterize.min.js"></script>
-<script>
-setTimeout(() => {
-window.initializePlayground({local: false})
-}, 1)
-</script>
diff --git a/docs/src/assets/js/playground.js b/docs/src/assets/js/playground.js
index e0ed6839..babb2afb 100644
--- a/docs/src/assets/js/playground.js
+++ b/docs/src/assets/js/playground.js
@@ -1,5 +1,3 @@
-import { Parser, Language } from './tree-sitter.js';
-
 function initializeLocalTheme() {
   const themeToggle = document.getElementById('theme-toggle');
   if (!themeToggle) return;
@@ -62,7 +60,9 @@ function initializeCustomSelect({ initialValue = null, addListeners = false }) {
   }
 }
 
-export async function initializePlayground(opts) {
+window.initializePlayground = async (opts) => {
+  const { Parser, Language } = window.TreeSitter;
+
   const { local } = opts;
   if (local) {
     initializeLocalTheme();