diff --git a/scripts/code-server/main.py b/scripts/code-server/main.py
index 88619f8..d91f1dc 100644
--- a/scripts/code-server/main.py
+++ b/scripts/code-server/main.py
@@ -1,7 +1,8 @@
 import os, sys
 
 with open("../nginx-site-template.conf", "r") as f:
-    template = f.read().replace("#serverNameVar", f"{sys.argv[1]}").replace("#http://127.0.0.1:8080/", f"http://172.30.3.1:8443/")
+    template = f.read().replace("#serverNameVar", f"{sys.argv[1]}").replace("#http://127.0.0.1:8080/", f"http://172.30.3.1:8443/") \
+        .replace("proxy_set_header Host $host;", "proxy_set_header Host $host;\nproxy_set_header Upgrade $http_upgrade;\nproxy_set_header Connection \"upgrade\";")
 
 with open("/etc/nginx/sites-available/code-server.conf", "w") as f:
     f.write(template)