diff --git a/build_kernel.sh b/build_kernel.sh
index 0ea4802..a131e59 100755
--- a/build_kernel.sh
+++ b/build_kernel.sh
@@ -24,6 +24,12 @@ 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
+
+# it needs more adaptation
+if [ "$5" == "arm" ];then
+    cp ${2}/kernel/OBJ/${8}/arch/arm/boot/uImage ${3}/uImage
+    cp ${2}/kernel/OBJ/${8}/arch/arm/boot/zImage-dtb ${3}/zImage-dtb
+elif [ "$5" == "arm64" ];then
+    cp ${2}/kernel/OBJ/${8}/arch/arm64/boot/Image ${3}/Image
+fi
 popd