diff --git a/ansible/roles/rad_setup_instructlab/tasks/50-dev-tools-setup.yml b/ansible/roles/rad_setup_instructlab/tasks/50-dev-tools-setup.yml
index 7ec8074abab..528ef60c9f6 100644
--- a/ansible/roles/rad_setup_instructlab/tasks/50-dev-tools-setup.yml
+++ b/ansible/roles/rad_setup_instructlab/tasks/50-dev-tools-setup.yml
@@ -156,10 +156,8 @@
         reboot_timeout: 180
       async: 1
       poll: 0
-    
-    - name: Wait for connection after reboot
-      wait_for_connection:
-        delay: 120
+      become: true
+      become_user: root
 
   become: true
   become_user: "{{ setup_instructlab_user }}"