diff --git a/scripts/initramfs.sh b/scripts/initramfs.sh
index 151f7b0a..49b9cede 100755
--- a/scripts/initramfs.sh
+++ b/scripts/initramfs.sh
@@ -2,11 +2,4 @@
 
 set -oue pipefail
 
-if [[ "${AKMODS_FLAVOR}" == "surface" ]]; then
-    KERNEL_NAME=kernel-surface
-else
-    KERNEL_NAME=kernel
-fi
-
-KERNEL="$(rpm -q "${KERNEL_NAME}" --queryformat '%{FEDORA_VERSION}-%{RELEASE}.%{ARCH}')"
-/usr/libexec/rpm-ostree/wrapped/dracut --no-hostonly --kver "${KERNEL}" --reproducible -v --add ostree -f /tmp/dracut
\ No newline at end of file
+/usr/libexec/rpm-ostree/wrapped/dracut --no-hostonly --kver "$(rpm -qa | grep -P 'kernel-(|'"$KERNEL_SUFFIX"'-)(\d+\.\d+\.\d+)' | sed -E 's/kernel-(|'"$KERNEL_SUFFIX"'-)//')" --reproducible -v --add ostree -f /tmp/dracut
\ No newline at end of file