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