diff --git a/builder/core/initramfs_builder.py b/builder/core/initramfs_builder.py
index e87785b6..70371538 100644
--- a/builder/core/initramfs_builder.py
+++ b/builder/core/initramfs_builder.py
@@ -68,7 +68,7 @@ def copy_kernel_image(rootfs_path, boot_path, target_architecture, iso_build_con
         raise ValueError("Could not obtain kernel version.")
     kernel_file_suffix = kernel_versions['version_final_part']
 
-    target_arch = iso_build_config.get('target_arch', 'x86_64').lower()
+    target_arch = target_architecture.lower() 
 
     if target_arch.startswith('i686') or target_arch == 'x86_64':
         kernel_filename_in_boot = "vmlinuz"