diff --git a/de/heat-templates/scripts/functions.sh b/de/heat-templates/scripts/functions.sh
index f09e6ea..3f838b6 100644
--- a/de/heat-templates/scripts/functions.sh
+++ b/de/heat-templates/scripts/functions.sh
@@ -26,6 +26,7 @@
 DOCKER_EE_RELEASE=${DOCKER_EE_RELEASE:-stable-19.03}
 DOCKER_EE_PACKAGES=${DOCKER_EE_PACKAGES:-'docker-ee'}
 BINARY_BASE_URL=${BINARY_BASE_URL:-"http://binary.mirantis.com"}
+CODENAME=$(cat /etc/*release | awk -F '=' '/_CODENAME/ {print $2}' | head -1)
 
 ### COMMON FUNCTIONS ###
 function get_interface_prefix {
@@ -192,7 +193,12 @@
     local pkg_list="apt-transport-https ca-certificates curl software-properties-common jq unzip atop iptables-persistent socat ntp rng-tools open-iscsi nfs-common"
 
     if [[ "${CONFIGURE_HUGE_PAGES}" == true ]]; then
-        pkg_list="$pkg_list dpdk-igb-uio-dkms"
+        # Since jammy the igb_uio kernel module contains in the dpdk-kmods-dkms package
+        if [[ ${CODENAME} == "bionic" ]] || [[ ${CODENAME} == "focal" ]]; then
+            pkg_list="$pkg_list dpdk-igb-uio-dkms"
+        else
+            pkg_list="$pkg_list dpdk-kmods-dkms"
+        fi
         # linux-image-extra-$(uname -r) provides uio_pci_generic module, used by TF DPDK as default driver
         if [[ "${TUNGSTENFABRIC_ENABLED,,}" == true ]]; then
             pkg_list="$pkg_list linux-modules-extra-$(uname -r)"