use become

handlers/main.yml

@@ -17,14 +17,10 @@
   listen:
     - "host_reboot"
 
-- name: Wait for host
-  wait_for:
-    host: "{{ ansible_ssh_host }}"
-    port: 22
+- name: Waiting for host startup
+  wait_for_connection:
     delay: 5
     timeout: 300
-  connection: local
-  become: False
   listen:
     - "network_restart"
     - "host_reboot"

tasks/config.yml

@@ -1,4 +1,6 @@
 ---
+- name: Setup network
+  block:
   - name: Set hostname
     template:
       src: etc/hostname.j2
@@ -42,3 +44,4 @@
       owner: root
       group: root
       mode: 0644
+  become: True