diff --git a/roles/os-updates/tasks/upgrade_packages.yml b/roles/os-updates/tasks/upgrade_packages.yml
index cd0f54f..d41c2ec 100644
--- a/roles/os-updates/tasks/upgrade_packages.yml
+++ b/roles/os-updates/tasks/upgrade_packages.yml
@@ -7,17 +7,16 @@
 
 - name: Check if a kernel update is available
   shell: |
-    dpkg -l | grep -E 'linux-image-[0-9]' | awk '{print $2}' | sort | tail -n 1
+    dpkg -l | grep -E '^ii' | grep 'linux-image-[0-9]' | awk '{print $2}' | sort | tail -n 1
   register: latest_kernel
 
 - name: Check if running kernel matches the latest installed kernel
-  shell: uname -r | grep -c "{{ latest_kernel.stdout }}"
+  shell: |
+    echo "{{ latest_kernel.stdout }}" | grep -c $(uname -r)
   register: kernel_match
   changed_when: false
 
 - name: Mark reboot required if a new kernel is installed
   set_fact:
     reboot_required: "yes"
-  notify: 
-    - Reboot system
   when: kernel_match.stdout == "0"
\ No newline at end of file