Refresh HWE kernel, and extra repo

Change-Id: Id5fafc43fcfdcd40aa14d59acfbc387de1874310
diff --git a/common/ubuntu_base.sh b/common/ubuntu_base.sh
index 58d3855..fbb475e 100644
--- a/common/ubuntu_base.sh
+++ b/common/ubuntu_base.sh
@@ -47,10 +47,7 @@
 # Cleanup old kernels, ensure latest is installed via metapackage package
 if [ ! -f /tmp/no_install_kernel ]; then
     apt-get purge -y linux-image-* linux-headers-* | grep -v 'is not installed, so not removed'
-    # FIXME: temporary switch back to LTS kernel, till PROD-23185 will be fixed
-    #apt-get install -y linux-image-virtual-hwe-16.04 linux-image-extra-virtual-hwe-16.04
-    apt-get install -y linux-image-virtual-lts-xenial linux-image-extra-virtual-lts-xenial
-
+    apt-get install -y linux-image-virtual-hwe-16.04 linux-image-extra-virtual-hwe-16.04 linux-headers-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
     sed -i 's|GRUB_CMDLINE_LINUX=.*|GRUB_CMDLINE_LINUX="console=tty0 console=ttyS0,115200n8"|g' /etc/default/grub