diff --git a/mk_mmc.sh b/mk_mmc.sh
index ca46def..9ab6927 100755
--- a/mk_mmc.sh
+++ b/mk_mmc.sh
@@ -435,7 +435,7 @@ function boot_uenv_txt_template {
 
 	cat >> ${TEMPDIR}/bootscripts/netinstall.cmd <<-__EOF__
 		address_image=kernel_addr
-		address_initrd=INITRD_ADDR
+		address_initrd=initrd_addr
 
 		console=DICONSOLE
 
@@ -452,7 +452,7 @@ function boot_uenv_txt_template {
 
 	cat >> ${TEMPDIR}/bootscripts/normal.cmd <<-__EOF__
 		address_image=kernel_addr
-		address_initrd=INITRD_ADDR
+		address_initrd=initrd_addr
 
 		console=SERIAL_CONSOLE
 
@@ -579,7 +579,7 @@ function tweak_boot_scripts {
 	sed -i -e 's:kernel_addr:'$kernel_addr':g' ${TEMPDIR}/bootscripts/${ALL}
 
 	#Set initrd boot address
-	sed -i -e 's:INITRD_ADDR:'$INITRD_ADDR':g' ${TEMPDIR}/bootscripts/${ALL}
+	sed -i -e 's:initrd_addr:'$initrd_addr':g' ${TEMPDIR}/bootscripts/${ALL}
 
 	#Set the Serial Console
 	sed -i -e 's:SERIAL_CONSOLE:'$SERIAL_CONSOLE':g' ${TEMPDIR}/bootscripts/${ALL}
@@ -1149,6 +1149,7 @@ cp -v "${DIR}/dl/${DISTARCH}/${ACTUAL_DEB_FILE}" ${TEMPDIR}/disk/
 			format=1.0
 			board=${BOOTLOADER}
 			kernel_addr=${kernel_addr}
+			initrd_addr=${initrd_addr}
 
 		__EOF__
 
@@ -1428,7 +1429,7 @@ function is_omap {
 	SUBARCH="omap"
 
 	kernel_addr="0x80300000"
-	INITRD_ADDR="0x81600000"
+	initrd_addr="0x81600000"
 
 	ZRELADD="0x80008000"
 
@@ -1616,7 +1617,7 @@ function check_uboot_type {
 		USE_ZIMAGE=1
 		ZRELADD="0x90008000"
 		kernel_addr="0x90800000"
-		INITRD_ADDR="0x92100000"
+		initrd_addr="0x92100000"
 		BETA_KERNEL=1
 		SERIAL_MODE=1
 		;;
@@ -1630,7 +1631,7 @@ function check_uboot_type {
 		USE_ZIMAGE=1
 		ZRELADD="0x70008000"
 		kernel_addr="0x70800000"
-		INITRD_ADDR="0x72100000"
+		initrd_addr="0x72100000"
 		BETA_KERNEL=1
 		SERIAL_MODE=1
 		;;