diff --git a/charts/netbox/templates/statefulset.yaml b/charts/netbox/templates/statefulset.yaml
index e7974e5..eccb2bc 100644
--- a/charts/netbox/templates/statefulset.yaml
+++ b/charts/netbox/templates/statefulset.yaml
@@ -19,24 +19,24 @@ spec:
 {{- end }}
   serviceName: 'netbox'
 {{ include "netbox.common" . | indent 2 }}
-{{- if .Values.persistence.enabled }}
-{{- if .Values.persistence.customClaim }}
-    {{ .Values.persistence.customClaim | toYaml | nindent 4 }}
+{{- if .Values.statefulSet.persistence.enabled }}
+{{- if .Values.statefulSet.persistence.customClaim }}
+    {{ .Values.statefulSet.persistence.customClaim | toYaml | nindent 4 }}
 {{- else }}
   volumeClaimTemplates:
   - spec:
       accessModes:
-      {{- range .Values.persistence.accessModes }}
+      {{- range .Values.statefulSet.persistence.accessModes }}
         - {{ . | quote }}
       {{- end }}
       resources:
         requests:
-          storage: "{{.Values.persistence.size }}"
-    {{- if .Values.persistence.storageClass }}
-    {{- if (eq "-" .Values.server.persistence.storageClass) }}
+          storage: "{{.Values.statefulSet.persistence.size }}"
+    {{- if .Values.statefulSet.persistence.storageClass }}
+    {{- if (eq "-" .Values.statefulSet.persistence.storageClass) }}
       storageClassName: ""
     {{- else }}
-      storageClassName: "{{ .Values.persistence.storageClass }}"
+      storageClassName: "{{ .Values.statefulSet.persistence.storageClass }}"
     {{- end }}
     {{- end }}
     metadata: