added restarting the node.
diff --git a/tasks/main.yml b/tasks/main.yml
index 49fd82c..02f440a 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -24,3 +24,8 @@
   when: "{{ ansible_selinux.status | lower != 'disabled' }}"
   async: 45
   poll: 0
+
+- name: Waiting for node to return
+  wait_for:
+    state=started
+    port=22