1 files changed, 5 insertions, 1 deletions
diff --git a/kexec.nix b/kexec.nix
index 0487197..4f14e01 100644
--- a/kexec.nix
+++ b/kexec.nix
@@ -11,7 +11,11 @@
   system.build = {
     image = pkgs.runCommand "image" { buildInputs = [ pkgs.nukeReferences ]; } ''
       mkdir $out
-      cp ${config.system.build.kernel}/bzImage $out/kernel
+      image=${builtins.getAttr config.nixpkgs.system {
+        aarch64-linux = "${config.system.build.kernel}/Image";
+        x86_64-linux = "${config.system.build.kernel}/bzImage";
+      }}
+      cp "$image" $out/kernel
       cp ${config.system.build.netbootRamdisk}/initrd $out/initrd
       nuke-refs $out/kernel
     '';