diff --git a/build_kernel.sh b/build_kernel.sh
index 0ea4802e031a7dc7db9af76c594d4ca534f79ccc..a7ef1998bda3b767bf594a06b452d34cd4e96728 100755
--- a/build_kernel.sh
+++ b/build_kernel.sh
@@ -24,6 +24,10 @@ pushd ${1}
 ./kernel_module_build.sh ${2} ${4} ${5} ${6} ${7} ${8}
 mkdir -p ${3}
 rm -rf ${3}/../../../kernel.timestamp
-cp ${2}/kernel/OBJ/${8}/arch/arm/boot/uImage ${3}/uImage
-cp ${2}/kernel/OBJ/${8}/arch/arm/boot/zImage-dtb ${3}/zImage-dtb
+if [ "${5}" == "arm" ];then
+    cp ${2}/kernel/OBJ/${8}/arch/${5}/boot/uImage ${3}/uImage
+    cp ${2}/kernel/OBJ/${8}/arch/${5}/boot/zImage-dtb ${3}/zImage-dtb
+elif [ "${5}" == "arm64" ];then
+    cp ${2}/kernel/OBJ/${8}/arch/${5}/boot/Image ${3}/Image
+fi
 popd