diff --git a/tests/integration/kubernetes/k8s-oom.bats b/tests/integration/kubernetes/k8s-oom.bats
index 52c0eeee3a..d2c9de8ab0 100644
--- a/tests/integration/kubernetes/k8s-oom.bats
+++ b/tests/integration/kubernetes/k8s-oom.bats
@@ -24,7 +24,12 @@ setup() {
 	kubectl wait --for=condition=Ready --timeout=$timeout pod "$pod_name"
 
 	# Check if OOMKilled
-	cmd="kubectl get pods "$pod_name" -o jsonpath='{.status.containerStatuses[0].state.terminated.reason}' | grep OOMKilled"
+    container_name=$(kubectl get pod "$pod_name" -o jsonpath='{.status.containerStatuses[0].name}')
+    if [[ $container_name == "oom-test" ]]; then
+        cmd="kubectl get pods "$pod_name" -o jsonpath='{.status.containerStatuses[0].state.terminated.reason}' | grep OOMKilled"
+    else
+        cmd="kubectl get pods "$pod_name" -o jsonpath='{.status.containerStatuses[1].state.terminated.reason}' | grep OOMKilled"
+    fi
 
 	waitForProcess "$wait_time" "$sleep_time" "$cmd"
 
diff --git a/tests/integration/kubernetes/runtimeclass_workloads/pod-oom.yaml b/tests/integration/kubernetes/runtimeclass_workloads/pod-oom.yaml
index 26badb3bc8..c0316616f7 100644
--- a/tests/integration/kubernetes/runtimeclass_workloads/pod-oom.yaml
+++ b/tests/integration/kubernetes/runtimeclass_workloads/pod-oom.yaml
@@ -17,6 +17,16 @@ spec:
       name: oom-test
       command: ["/bin/sh"]
       args: ["-c", "sleep 2; stress --vm 2 --vm-bytes 500M --timeout 30s"]
+      resources:
+        limits:
+          memory: 400Mi
+        requests:
+          memory: 400Mi
+    - image: quay.io/kata-containers/sysbench-kata:latest
+      imagePullPolicy: IfNotPresent
+      name: not-oom
+      command: ["/bin/sh"]
+      args: ["-c", "sleep inf"]
       resources:
         limits:
           memory: 500Mi