updated logic

tasks/main.yml

diff --git a/tasks/main.yml b/tasks/main.yml
index 1a7245f..dd0d46b 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -8,12 +8,7 @@
 
 - name: Waiting for server to complete restarting
   become: no
-  delegate_to: 127.0.0.1
-  wait_for:
-    host: '{{ ansible_host }}'
-    state: started
-    delay: "{{ start_check_delay | d(15) }}"
-    timeout: '{{ server_restart_timeout | d(45) }}'
+  local_action: wait_for host='{{ ansible_host }}' state=started delay="{{ start_check_delay | d(15) }}" timeout='{{ server_restart_timeout | d(45) }}'
 
 - name: Report server started
   debug: