diff --git a/assets/reload.js b/assets/reload.js
index 5d4a3f4..c95d9bd 100644
--- a/assets/reload.js
+++ b/assets/reload.js
@@ -9,16 +9,16 @@ function reload() {
     });
 }
 
-async function pollServer() {
-  try {
-    console.log("Polling server...");
-    let response = await fetch("http://localhost:12345/");
-    reload();
-    pollServer();
-  } catch (err) {
-    console.log(err);
-    console.log("Server not ready. Reload the page with the server running.");
-  }
+function pollServer() {
+  fetch("http://localhost:12345/")
+    .then(() => {
+      reload();
+      pollServer();
+    })
+    .catch((err) => {
+      console.log(err);
+      console.log("Server not ready. Reload the page with the server running.");
+    });
 }
 
 pollServer();
diff --git a/lib/local-server.rb b/lib/local-server.rb
index 374aed3..1e03802 100644
--- a/lib/local-server.rb
+++ b/lib/local-server.rb
@@ -27,6 +27,7 @@ def watch_files
     @thread = Thread.new(filewatcher) do |fw|
       fw.watch do |change|
         puts "Change detected: #{change}" if opts.verbose
+        # TODO: change this to use exposed methods
         generator.send(:write_html)
         @needs_reload = true
       end