diff --git a/playbooks/update.yml b/playbooks/update.yml
index b7b89c6..70580f1 100644
--- a/playbooks/update.yml
+++ b/playbooks/update.yml
@@ -2,6 +2,10 @@
 - name: 'Install updates through package management'
   hosts: 'all'
 
+  handlers:
+    - name: 'Reboot machine'
+      ansible.builtin.reboot:
+
   pre_tasks:
     - name: 'Update package cache if needed'
       become: true
@@ -20,3 +24,17 @@
       ansible.builtin.apt:
         autoclean: true
         autoremove: true
+
+    - name: 'Check if reboot is required'
+      ansible.builtin.stat:
+        path: '/var/run/reboot-required'
+      register: 'reboot_required_file'
+
+    - name: 'Reboot if required'
+      when: 'reboot_required_file.stat.exists'
+      ansible.builtin.assert:
+        that:
+          - true
+        quiet: true
+      changed_when: true
+      notify: 'Reboot machine'