diff --git a/apt_update_ubuntu.yml b/apt_update_ubuntu.yml
index 5e9d797..f8e1251 100644
--- a/apt_update_ubuntu.yml
+++ b/apt_update_ubuntu.yml
@@ -7,3 +7,12 @@
         update_cache: yes
         upgrade: yes
         autoremove: yes
+
+    - name: Check if reboot required
+      stat:
+      path: /var/run/reboot-required
+      register: reboot_required_file
+
+    - name: Reboot if required
+      reboot:
+      when: reboot_required_file.stat.exists == true