diff --git a/arch/arm/mach-rockchip/make_fit_boot.sh b/arch/arm/mach-rockchip/make_fit_boot.sh
index 0c5993e455..8ad86be715 100755
--- a/arch/arm/mach-rockchip/make_fit_boot.sh
+++ b/arch/arm/mach-rockchip/make_fit_boot.sh
@@ -14,8 +14,10 @@ fi
 
 if [ "${COMPRESSION}" == "gzip" ]; then
 	gzip -k -f -9 ${srctree}/images/kernel
-	gzip -k -f -9 ${srctree}/images/ramdisk
 	SUFFIX=".gz"
+elif [ "${COMPRESSION}" == "lz4" ]; then
+	lz4c -9 -f ${srctree}/images/kernel > ${srctree}/images/kernel.lz4
+	SUFFIX=".lz4"
 else
 	COMPRESSION="none"
 	SUFFIX=
@@ -48,10 +50,8 @@ cat << EOF
 		};
 
 		kernel {
-EOF
-echo "			data = /incbin/(\"./images/kernel${SUFFIX}\");"
-echo "			compression = \"${COMPRESSION}\";"
-cat << EOF
+			data = /incbin/("./images/kernel${SUFFIX}");
+			compression = "${COMPRESSION}";
 			type = "kernel";
 			arch = "${ARCH}";
 			os = "linux";
@@ -63,10 +63,8 @@ cat << EOF
 		};
 
 		ramdisk {
-EOF
-echo "			data = /incbin/(\"./images/ramdisk${SUFFIX}\");"
-echo "			compression = \"${COMPRESSION}\";"
-cat << EOF
+			data = /incbin/("./images/ramdisk");
+			compression = "none";
 			type = "ramdisk";
 			arch = "${ARCH}";
 			os = "linux";