diff --git a/tests/integration/kubernetes/lib.sh b/tests/integration/kubernetes/lib.sh
index bfc4c3c3d9..c726a0c14c 100644
--- a/tests/integration/kubernetes/lib.sh
+++ b/tests/integration/kubernetes/lib.sh
@@ -340,10 +340,19 @@ set_container_command() {
 set_node() {
 	local yaml="$1"
 	local node="$2"
+	local kind
+	local spec
 	[ -n "$node" ] || return 1
 
+	kind="$(yq -r '.kind' "${yaml}")"
+	if [[ "${kind}" = "Job" ]]; then
+		spec=".spec.template.spec.nodeName"
+	else
+		spec=".spec.nodeName"
+	fi
+
   yq -i \
-    ".spec.nodeName = \"$node\"" \
+    "${spec} = \"$node\"" \
     "${yaml}"
 }