updated logic

tasks/main.yml

diff --git a/tasks/main.yml b/tasks/main.yml
index aa1f1db..56c5545 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -10,7 +10,7 @@
   become: no
   local_action: wait_for
   args:
-    host: '{{ inventory_hostname }}'
+    host: '{{ ansible_host }}'
     port: 22
     state: started
     delay: "{{ server_restart_delay | d(15) }}"