diff --git a/tasks/main.yml b/tasks/main.yml
index 02f440a..2364617 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -17,15 +17,20 @@
     msg: 'SELinux MUST be disabled, selinux has been updated, restarting node now...'
   when: "{{ ansible_selinux.status | lower != 'disabled' }}"
 
-- name: Restarting node now
-  become: true
-  ignore_errors: yes
-  shell: 'reboot now'
-  when: "{{ ansible_selinux.status | lower != 'disabled' }}"
-  async: 45
-  poll: 0
+- block:
 
-- name: Waiting for node to return
-  wait_for:
-    state=started
-    port=22
+  - name: Restarting node now
+    become: true
+    ignore_errors: yes
+    shell: 'reboot now'
+    when: "{{ ansible_selinux.status | lower != 'disabled' }}"
+    async: 45
+    poll: 0
+
+  rescue:
+  - name: Waiting for node to return
+    local_action:
+      wait_for:
+        state: started
+        host: '{{ inventory_host }}'
+        port: 22