diff --git a/ansible/tasks/sgx_setup.yml b/ansible/tasks/sgx_setup.yml
index 98c9eca..6ca798e 100644
--- a/ansible/tasks/sgx_setup.yml
+++ b/ansible/tasks/sgx_setup.yml
@@ -6,17 +6,21 @@
 - name: "Set WASM VM env. variable to SGX"
   shell: "echo 'export PATH=$PATH:/home/{{ ansible_user }}/.local/bin' >> /home/{{ ansible_user }}/.bashrc"
 
-- name: "Upgrade to 6.x kernel for EDMM features"
+- name: "Update apt cache"
+  become: yes
+  apt:
+    update_cache: yes
+
+- name: "Install linux recommends"
   become: yes
-  tasks:
-    - name: "Update apt cache"
-      apt:
-        update_cache: yes
+  shell: "apt install --install-recommends -y linux-generic"
 
-    - name: "Install the specific kernel version"
-      apt:
-        name: linux-image-unsigned-6.5.0-1011-azure
-        state: present
+- name: "Upgrade to 6.x kernel for EDMM features"
+  become: yes
+  apt:
+    name: linux-image-6.5.0-1025-azure
+  environment:
+    DEBIAN_FRONTEND: noninteractive
 
 - name: "Reboot to pick-up latest kernel"
   reboot: