CORD-1894 Cleanup for HW pod install

Change-Id: I685861a1bdcbdc40aa993d5d9b0c01477fdde8c6
(cherry picked from commit 31385c19d676c72e0227af17c36dd5a0dee63da1)
diff --git a/roles/compute-node/tasks/networking.yml b/roles/compute-node/tasks/networking.yml
index 6411335..37b9617 100644
--- a/roles/compute-node/tasks/networking.yml
+++ b/roles/compute-node/tasks/networking.yml
@@ -410,7 +410,10 @@
     - reboot
 
 - name: Wait For Restart
-  local_action: wait_for port={{ ansible_port | default('22') }} host={{ inventory_hostname }} search_regex=OpenSSH delay=120 timeout=600 connect_timeout=15
+  wait_for_connection:
+    delay: 120
+    timeout: 600
+    connect_timeout: 15
   when: reboot_required
   tags:
     - interface_config