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 2cbde5c..29f970f 100644
--- a/target/linux/mediatek/mt7981/base-files/lib/upgrade/platform.sh
+++ b/target/linux/mediatek/mt7981/base-files/lib/upgrade/platform.sh
@@ -1,10 +1,10 @@
-RAMFS_COPY_BIN='mkfs.f2fs blkid blockdev'
+RAMFS_COPY_BIN='mkfs.f2fs blkid blockdev fw_printenv fw_setenv'
 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"