diff --git a/deploy/helm/seaweedfs-csi-driver/templates/deployment.yaml b/deploy/helm/seaweedfs-csi-driver/templates/deployment.yaml
index edbc143..b955597 100644
--- a/deploy/helm/seaweedfs-csi-driver/templates/deployment.yaml
+++ b/deploy/helm/seaweedfs-csi-driver/templates/deployment.yaml
@@ -16,6 +16,9 @@ spec:
       priorityClassName: system-cluster-critical
       serviceAccountName: {{ template "seaweedfs-csi-driver.name" . }}-controller-sa
       affinity:
+        {{- if and .Values.controller.affinity .Values.controller.affinity.nodeAffinity }}
+        nodeAffinity: {{ toYaml .Values.controller.affinity.nodeAffinity | nindent 10 }}
+        {{- end }}
         {{- if and .Values.controller.affinity .Values.controller.affinity.podAffinity }}
         podAffinity: {{ toYaml .Values.controller.affinity.podAffinity | nindent 10 }}
         {{- end }}