diff --git a/target/linux/mediatek/filogic/base-files/lib/upgrade/platform.sh b/target/linux/mediatek/filogic/base-files/lib/upgrade/platform.sh
index 0a6f892a1..6642318f8 100755
--- a/target/linux/mediatek/filogic/base-files/lib/upgrade/platform.sh
+++ b/target/linux/mediatek/filogic/base-files/lib/upgrade/platform.sh
@@ -1,5 +1,16 @@
 REQUIRE_IMAGE_METADATA=1
 
+asus_initial_setup()
+{
+	# initialize UBI if it's running on initramfs
+	[ "$(rootfs_type)" = "tmpfs" ] || return 0
+
+	ubirmvol /dev/ubi0 -N rootfs
+	ubirmvol /dev/ubi0 -N rootfs_data
+	ubirmvol /dev/ubi0 -N jffs2
+	ubimkvol /dev/ubi0 -N jffs2 -s 0x3e000
+}
+
 platform_do_upgrade() {
 	local board=$(board_name)
 
@@ -84,4 +95,14 @@ platform_copy_config() {
 		emmc_copy_config
 		;;
 	esac
+ }
+ 
+platform_pre_upgrade() {
+	local board=$(board_name)
+
+	case "$board" in
+	asus,tuf-ax4200)
+		asus_initial_setup
+		;;
+	esac
 }