diff --git a/nginx.sh b/nginx.sh
index fbce693..87011ab 100755
--- a/nginx.sh
+++ b/nginx.sh
@@ -468,7 +468,7 @@ proxy() { local service="$1" location="$2" header="${3:-""}" \
 \
 '"$([[ $sockets != "no" ]] && echo '        ## Required for websockets\
         proxy_http_version 1.1;\
-        proxy_set_header Connection "upgrade";\
+        proxy_set_header Connection $http_connection;\
         proxy_set_header Upgrade $http_upgrade;\
         proxy_read_timeout 3600s;\
         proxy_send_timeout 3600s;')"'\
@@ -724,4 +724,4 @@ elif ps -ef | egrep -v 'grep|nginx.sh' | grep -q nginx; then
     echo "Service already running, please restart container to apply changes"
 else
     exec nginx -g "daemon off;"
-fi
\ No newline at end of file
+fi