diff --git a/test/e2e/policies.test-suite/balloons/n4c16/test22-isolcpus/code.var.sh b/test/e2e/policies.test-suite/balloons/n4c16/test22-isolcpus/code.var.sh
index 7045c8553..36c86a0fc 100644
--- a/test/e2e/policies.test-suite/balloons/n4c16/test22-isolcpus/code.var.sh
+++ b/test/e2e/policies.test-suite/balloons/n4c16/test22-isolcpus/code.var.sh
@@ -1,6 +1,5 @@
 reboot-node() {
-    vm-reboot
-    timeout=600 host-wait-vm-ssh-server $OUTPUT_DIR
+    timeout=600 vm-reboot
 }
 
 restart-kubelet() {