diff --git a/linux_updates.yaml b/linux_updates.yaml
index 38a189f..f7a1329 100644
--- a/linux_updates.yaml
+++ b/linux_updates.yaml
@@ -37,10 +37,12 @@
     ansible.builtin.apt:
       autoremove: yes
 
-  - name: Reboot when packages were updated
-    reboot:
-      post_reboot_delay: 60
-    when: apt_update_status.changed
+  - name: Reboot if required
+    ansible.builtin.reboot:
+      reboot_timeout: 600 # Adjust timeout as needed
+    when: reboot_required_file.stat.exists
+    vars:
+      reboot_required_file: /var/run/reboot-required
 
 #- hosts: apkhosts
 #  tasks: