diff --git a/roles/common-prep/tasks/main.yml b/roles/common-prep/tasks/main.yml
index 0e5474d..a7ffc3b 100644
--- a/roles/common-prep/tasks/main.yml
+++ b/roles/common-prep/tasks/main.yml
@@ -3,6 +3,9 @@
 
 - name: Upgrade system to current using apt
   apt: update_cache=yes upgrade=dist
+  notify:
+    - restart machine
+    - wait for machine
 
 - name: Install standard packages
   apt:  pkg={{ item }} state=present
@@ -14,3 +17,4 @@
   lineinfile: dest=/etc/vim/vimrc
     regexp="^\s*syntax on"
     line="syntax on"
+