diff --git a/helm/templates/kafka/kafka-cluster.yaml b/helm/templates/kafka/kafka-cluster.yaml
index f1a58077a78865c624706531b58c3150feeb83ae..ee2bd0f1c529f98a9d838a704c6f277e9106f439 100644
--- a/helm/templates/kafka/kafka-cluster.yaml
+++ b/helm/templates/kafka/kafka-cluster.yaml
@@ -30,14 +30,9 @@ spec:
         configMapKeyRef:
           name: {{ template "theodolite.fullname" . }}-kafka-metrics
           key: kafka-metrics-config.yml
-    {{- with .Values.strimzi.kafka.nodeSelectorTerms}}
+    {{- with .Values.strimzi.kafka.template}}
     template:
-      pod:
-        affinity:
-          nodeAffinity:
-            requiredDuringSchedulingIgnoredDuringExecution:
-              nodeSelectorTerms:
-                {{- toYaml . | nindent 16 }}
+      {{- toYaml . | nindent 6 }}
     {{- end}}
     {{- with .Values.strimzi.kafka.resources}}
     resources:
@@ -52,14 +47,9 @@ spec:
     {{- end }}
     storage:
       type: ephemeral
-    {{- with .Values.strimzi.zookeeper.nodeSelectorTerms}}
+    {{- with .Values.strimzi.kafka.template}}
     template:
-      pod:
-        affinity:
-          nodeAffinity:
-            requiredDuringSchedulingIgnoredDuringExecution:
-              nodeSelectorTerms:
-                {{- toYaml . | nindent 16 }}
+      {{- toYaml . | nindent 6 }}
     {{- end}}
 
   kafkaExporter: {}