diff --git a/memphis/templates/memphis_statefulset.yaml b/memphis/templates/memphis_statefulset.yaml
index 453496d..2bff2ac 100644
--- a/memphis/templates/memphis_statefulset.yaml
+++ b/memphis/templates/memphis_statefulset.yaml
@@ -279,6 +279,10 @@ spec:
               key: {{ .key }}
         {{- end }}
         {{- end }}
+        {{- with .Values.websocket }}
+        - name: WS_HOST
+          value: {{ .host }}:{{ .port }}
+        {{- end }}
         - name: USER_PASS_BASED_AUTH
           value: {{ .Values.memphis.configFile.userPassBasedAuth | quote }}
         - name: LOGS_RETENTION_IN_DAYS
diff --git a/memphis/values.yaml b/memphis/values.yaml
index bc54e97..b2d5421 100644
--- a/memphis/values.yaml
+++ b/memphis/values.yaml
@@ -480,13 +480,11 @@ auth:
 websocket:
   enabled: true
   port: 7770
+  # Websocket URL configuration.
+  host: localhost
   noTLS: true
   sameOrigin: false
   allowedOrigins: []
-  #auth:
-  #  token: memphis
-  #  user: ""
-  #  pass: ""
 appProtocol:
   enabled: false