diff --git a/roles/packages/tasks/main.yml b/roles/packages/tasks/main.yml
index de034f1..23b9557 100644
--- a/roles/packages/tasks/main.yml
+++ b/roles/packages/tasks/main.yml
@@ -9,7 +9,7 @@
 
 - name: Remove Ubuntu packages
   apt:
-    name: ['cloud-init', 'modemmanager', 'snapd', 'modemmanager']
+    name: ['cloud-init', 'modemmanager', 'snapd', 'modemmanager', 'needrestart']
     state: absent
   when: ansible_facts['os_family'] == "Debian"
 
@@ -26,6 +26,7 @@
     replace: 'MODULES=dep'
   notify: 
   - Make initramfs
+  when: ansible_facts['os_family'] == "Debian"
 
 - name: Install CentOS epel
   yum: