diff --git a/default.conf b/default.conf
index c4b2947..0d58977 100644
--- a/default.conf
+++ b/default.conf
@@ -7,6 +7,7 @@ server {
     proxy_set_header X-Real-IP          $remote_addr;
     proxy_set_header X-Forwarded-For    $proxy_add_x_forwarded_for;
     proxy_set_header X-Forwarded-Proto  $scheme;
+    proxy_set_header X-Forwarded-Scheme $scheme;
     proxy_set_header REMOTE-HOST        $remote_addr;
 
     proxy_redirect                      off;
@@ -15,5 +16,10 @@ server {
         proxy_pass                          http://nodeui:3000;  # must be same name as service in docker-compose
         proxy_read_timeout                  1m;
         proxy_connect_timeout               1m;
+
+        # Enable WebSocket connections
+        proxy_http_version 1.1;
+        proxy_set_header Upgrade $http_upgrade;
+        proxy_set_header Connection "upgrade";
     }
 }