diff --git a/ubuntu-16.04/scripts/base.sh b/ubuntu-16.04/scripts/base.sh
index 07261d2..dea0a72 100644
--- a/ubuntu-16.04/scripts/base.sh
+++ b/ubuntu-16.04/scripts/base.sh
@@ -13,9 +13,10 @@
 curl -sf http://apt-mk.mirantis.com/public.gpg | apt-key add -
 
 # Cleanup old kernels, ensure latest is installed via virtual package
-#apt-get purge -y linux-image-* linux-headers-*
+apt-get purge -y linux-image-* linux-headers-*
 if [ ! -f /tmp/no_install_kernel ]; then
-#    apt-get install -y linux-generic-lts-vivid
+    # Use HWE kernel
+    apt-get install -y linux-image-generic-hwe-16.04
 
     # Update grub cmdline
     sed -i 's|GRUB_CMDLINE_LINUX_DEFAULT=.*|GRUB_CMDLINE_LINUX_DEFAULT="console=tty0 console=ttyS0,115200n8"|g' /etc/default/grub