diff --git a/roles/rke2/tasks/setup_host.yml b/roles/rke2/tasks/setup_host.yml
index 4e1f5c0711ffde24f69b4c57ad206ab888648a78..8ad8bdec3ac0257d2ab94a961934b9c5f8eef285 100644
--- a/roles/rke2/tasks/setup_host.yml
+++ b/roles/rke2/tasks/setup_host.yml
@@ -42,7 +42,7 @@
 - name: update package cache
   apt:
     update_cache: yes
-#  when: dist_upgrade
+  when: dist_upgrade
 
 
 - name: upgrade packages
@@ -50,10 +50,11 @@
     name: "*"
     state: latest
   register: update_packages
-#  when: dist_upgrade
+  when: dist_upgrade
 
 - name: reboot vm
   throttle: 1
   reboot:
     reboot_timeout: 300
+    post_reboot_delay: 60
   when: update_packages.changed 
\ No newline at end of file