diff --git a/cli/src/web_ui.html b/cli/src/web_ui.html
index 62c23f3d..093b09ba 100644
--- a/cli/src/web_ui.html
+++ b/cli/src/web_ui.html
@@ -7,7 +7,7 @@
 </head>
 
 <body>
-  <div id="playground-container">
+  <div id="playground-container" style="visibility: hidden;">
     <header>
       <div class=header-item>
         <bold>THE_LANGUAGE_NAME</bold>
@@ -118,7 +118,12 @@
       margin-right: 30px;
     }
 
-    .CodeMirror {
+    #playground-container .CodeMirror {
+      position: absolute;
+      top: 0;
+      bottom: 0;
+      left: 0;
+      right: 0;
       height: 100%;
     }
 
diff --git a/script/serve-docs b/script/serve-docs
index f3a80a15..2e9b258e 100755
--- a/script/serve-docs
+++ b/script/serve-docs
@@ -9,14 +9,14 @@ bundle exec ruby <<RUBY &
   require "listen"
 
   def copy_wasm_files
-    `cp $root/target/release/*.{js,wasm} $root/docs/assets/js/`
+    `cp $root/lib/binding_web/tree-sitter.{js,wasm} $root/docs/assets/js/`
   end
 
   puts "Copying WASM files to docs folder..."
   copy_wasm_files
 
   puts "Watching release directory"
-  listener = Listen.to("$root/target/release", only: /^tree-sitter\.(js|wasm)$/, wait_for_delay: 2) do
+  listener = Listen.to("$root/lib/binding_web", only: /^tree-sitter\.(js|wasm)$/, wait_for_delay: 2) do
     puts "WASM files updated. Copying new files to docs folder..."
     copy_wasm_files
   end
