diff --git a/target/linux/mediatek/mt7981/base-files/lib/upgrade/platform.sh b/target/linux/mediatek/mt7981/base-files/lib/upgrade/platform.sh
index 6eb1e5f..bd38b22 100644
--- a/target/linux/mediatek/mt7981/base-files/lib/upgrade/platform.sh
+++ b/target/linux/mediatek/mt7981/base-files/lib/upgrade/platform.sh
@@ -1,5 +1,6 @@
 RAMFS_COPY_BIN='mkfs.f2fs blkid blockdev fw_printenv fw_setenv'
 RAMFS_COPY_DATA="/etc/fw_env.config /var/lock/fw_printenv.lock"
+
 platform_do_upgrade() {
 	local board=$(board_name)
 
@@ -48,4 +49,3 @@
 
 	return 0
 }
-
diff --git a/target/linux/mediatek/mt7988/base-files/lib/upgrade/platform.sh b/target/linux/mediatek/mt7988/base-files/lib/upgrade/platform.sh
index 2ee99c7..bd38b22 100644
--- a/target/linux/mediatek/mt7988/base-files/lib/upgrade/platform.sh
+++ b/target/linux/mediatek/mt7988/base-files/lib/upgrade/platform.sh
@@ -1,9 +1,15 @@
+RAMFS_COPY_BIN='mkfs.f2fs blkid blockdev fw_printenv fw_setenv'
+RAMFS_COPY_DATA="/etc/fw_env.config /var/lock/fw_printenv.lock"
+
 platform_do_upgrade() {
 	local board=$(board_name)
 
 	case "$board" in
 	*snand*)
-		nand_do_upgrade "$1"
+		ubi_do_upgrade "$1"
+		;;
+	*emmc*)
+		mtk_mmc_do_upgrade "$1"
 		;;
 	*)
 		default_do_upgrade "$1"
@@ -20,7 +26,8 @@
 	[ "$#" -gt 1 ] && return 1
 
 	case "$board" in
-	*snand*)
+	*snand* |\
+	*emmc*)
 		# tar magic `ustar`
 		magic="$(dd if="$1" bs=1 skip=257 count=5 2>/dev/null)"
 
@@ -42,4 +49,3 @@
 
 	return 0
 }
-