diff --git a/charts/open-webui/Chart.yaml b/charts/open-webui/Chart.yaml
index 7c00792..a660d20 100644
--- a/charts/open-webui/Chart.yaml
+++ b/charts/open-webui/Chart.yaml
@@ -1,7 +1,7 @@
 apiVersion: v2
 name: open-webui
-version: 3.0.1
-appVersion: "0.2.5"
+version: 3.0.2
+appVersion: "v0.3.4"
 
 home: https://www.openwebui.com/
 icon: https://raw.githubusercontent.com/open-webui/open-webui/main/static/favicon.png
diff --git a/charts/open-webui/templates/deployment.yaml b/charts/open-webui/templates/workload-manager.yaml
similarity index 97%
rename from charts/open-webui/templates/deployment.yaml
rename to charts/open-webui/templates/workload-manager.yaml
index f8a171f..8a20bb1 100644
--- a/charts/open-webui/templates/deployment.yaml
+++ b/charts/open-webui/templates/workload-manager.yaml
@@ -1,5 +1,9 @@
 apiVersion: apps/v1
+{{- if .Values.persistence.enabled }}
+kind: StatefulSet
+{{- else }}
 kind: Deployment
+{{- end }}
 metadata:
   name: {{ include "open-webui.name" . }}
   labels: