diff --git a/charts/redis-stack-server/templates/redis-stack-server.yaml b/charts/redis-stack-server/templates/redis-stack-server.yaml
index 5ce22d3..7a04701 100644
--- a/charts/redis-stack-server/templates/redis-stack-server.yaml
+++ b/charts/redis-stack-server/templates/redis-stack-server.yaml
@@ -28,6 +28,10 @@ spec:
       labels:
         app: "{{ .Values.name }}"
     spec:
+    {{- if .Values.redis_stack_server.affinity }}
+      affinity:
+    {{- toYaml .Values.redis_stack_server.affinity | nindent 8 }}
+    {{- end }}
       terminationGracePeriodSeconds: 10
       containers:
       - name: "{{ .Values.name }}"
@@ -47,8 +51,4 @@ spec:
       storageClassName: {{ .Values.redis_stack_server.storage_class }}
       resources:
         requests:
-          storage: {{ .Values.redis_stack_server.storage }}
-{{- if .Values.redis_stack_server.affinity }}
-      affinity:
-{{ toYaml .Values.redis_stack_server.affinity | indent 8 }}
-{{- end }}
+          storage: {{ .Values.redis_stack_server.storage }}
\ No newline at end of file