diff --git a/metal/roles/pxe-boot/tasks/wake.yml b/metal/roles/pxe-boot/tasks/wake.yml
index ebd8e736..e9eaa8d7 100644
--- a/metal/roles/pxe-boot/tasks/wake.yml
+++ b/metal/roles/pxe-boot/tasks/wake.yml
@@ -4,5 +4,8 @@
   delegate_to: localhost
 
 - name: Wait for the servers to comes up
-  wait_for_connection:
-    timeout: 600
+  wait_for:
+    host: '{{ ansible_host }}'
+    port: 22
+    search_regex: OpenSSH
+  connection: local