diff --git a/builder/core/initramfs_builder.py b/builder/core/initramfs_builder.py
index 0c153d6e..a2e483fe 100644
--- a/builder/core/initramfs_builder.py
+++ b/builder/core/initramfs_builder.py
@@ -164,7 +164,7 @@ def create_initramfs(rootfs_path, boot_path, target_architecture, xbps_commands,
     logger.info("=> Starting initramfs creation...")
     logger.info("=> Creating initramfs using dracut...")
 
-    kernel_versions = get_kernel_version_in_chroot(rootfs_path, iso_build_config)
+    kernel_versions = get_kernel_version_in_chroot(rootfs_path, target_architecture, iso_build_config)
     if not kernel_versions or 'version_final_part' not in kernel_versions:
         logger.error("=> ERROR: Could not obtain kernel version to create the initramfs.")
         raise ValueError("Could not obtain kernel version.")