diff --git a/tasks/main.yml b/tasks/main.yml
index 27dc385..290264c 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -22,7 +22,7 @@
   - name: Waiting for server to complete restarting
     become: no
     local_action:
-      wait_for host='{{ ansible_ssh_host }}' state=started delay="{{ start_check_delay | d(30) }}" timeout='{{ server_restart_timeout | d(45) }}'
+      wait_for host='{{ inventory_hostname }}' state=started delay="{{ start_check_delay | d(30) }}" timeout='{{ server_restart_timeout | d(45) }}'
 
   - name: Report server started
     debug: