diff --git a/roles/proxmox_lxc/tasks/main.yml b/roles/proxmox_lxc/tasks/main.yml
index 22d6251..53b22b9 100644
--- a/roles/proxmox_lxc/tasks/main.yml
+++ b/roles/proxmox_lxc/tasks/main.yml
@@ -12,10 +12,6 @@
   changed_when: false
   ignore_errors: yes
 
-- name: Display template check result
-  debug:
-    var: template_check
-
 - name: Create Gateway LXC container using correct template path
   shell: |
     pct create 221 \
@@ -37,9 +33,10 @@
     - "'221' not in container_check.stdout"
     - "template_check.rc == 0"
 
-- name: Wait for container to start
-  wait_for_connection:
+- name: Wait for container SSH port to be available
+  wait_for:
     host: 192.168.1.221
     port: 22
     timeout: 120
-    delay: 10
\ No newline at end of file
+    delay: 10
+  delegate_to: localhost
\ No newline at end of file