diff --git a/tests/integration/kubernetes/k8s-attach-handlers.bats b/tests/integration/kubernetes/k8s-attach-handlers.bats
index 10a7a0f199..fa38534ed2 100644
--- a/tests/integration/kubernetes/k8s-attach-handlers.bats
+++ b/tests/integration/kubernetes/k8s-attach-handlers.bats
@@ -15,28 +15,39 @@ setup() {
 	pod_name="handlers"
 
 	get_pod_config_dir
+	yaml_file="${pod_config_dir}/test-lifecycle-events.yaml"
+
+	# Create yaml
+	sed -e "s/\${nginx_version}/${nginx_image}/" \
+		"${pod_config_dir}/lifecycle-events.yaml" > "${yaml_file}"
+
+	# Add policy to yaml
+	policy_settings_dir="$(create_tmp_policy_settings_dir "${pod_config_dir}")"
+	display_message="cat /usr/share/message"
+	exec_command="sh -c ${display_message}"
+	add_exec_to_policy_settings "${policy_settings_dir}" "${exec_command}"
+	auto_generate_policy "${policy_settings_dir}" "${yaml_file}"
 }
 
 @test "Running with postStart and preStop handlers" {
-	# Create yaml
-	sed -e "s/\${nginx_version}/${nginx_image}/" \
-		"${pod_config_dir}/lifecycle-events.yaml" > "${pod_config_dir}/test-lifecycle-events.yaml"
-
 	# Create the pod with postStart and preStop handlers
-	kubectl create -f "${pod_config_dir}/test-lifecycle-events.yaml"
+	kubectl create -f "${yaml_file}"
 
 	# Check pod creation
 	kubectl wait --for=condition=Ready --timeout=$timeout pod $pod_name
 
 	# Check postStart message
-	display_message="cat /usr/share/message"
-	check_postStart=$(kubectl exec $pod_name -- sh -c "$display_message" | grep "Hello from the postStart handler")
+	check_postStart=$(kubectl exec $pod_name -- sh -c "$display_message")
+	echo "check_postStart=$check_postStart"
+	echo "$check_postStart" | grep "Hello from the postStart handler"
 }
 
 teardown(){
 	# Debugging information
 	kubectl describe "pod/$pod_name"
 
-	rm -f "${pod_config_dir}/test-lifecycle-events.yaml"
+	rm -f "${yaml_file}"
 	kubectl delete pod "$pod_name"
+
+	delete_tmp_policy_settings_dir "${policy_settings_dir}"
 }