diff --git a/patches/0017-ipq807x-add-the-Qualcomm-AX-target-support.patch b/patches/0017-ipq807x-add-the-Qualcomm-AX-target-support.patch
index 44827279e..f69bcbab2 100644
--- a/patches/0017-ipq807x-add-the-Qualcomm-AX-target-support.patch
+++ b/patches/0017-ipq807x-add-the-Qualcomm-AX-target-support.patch
@@ -1,4 +1,4 @@
-From d977d1240924a1ab3d283cd7bd91bcb3004e7949 Mon Sep 17 00:00:00 2001
+From 7b587e276da805d981596725d9eb5421d2d9c925 Mon Sep 17 00:00:00 2001
 From: John Crispin <john@phrozen.org>
 Date: Sat, 18 Jul 2020 08:53:44 +0200
 Subject: [PATCH 01/36] ipq807x: add the Qualcomm AX target support
@@ -20,7 +20,7 @@ Signed-off-by: John Crispin <john@phrozen.org>
  .../ipq807x/base-files/etc/init.d/bootcount   |    12 +
  .../linux/ipq807x/base-files/etc/init.d/wdt   |    14 +
  ...G4_v5.4.B-AQR_CIG_WIFI_ID44715_VER1673.cld |   Bin 0 -> 391170 bytes
- .../base-files/lib/upgrade/platform.sh        |    72 +
+ .../base-files/lib/upgrade/platform.sh        |    73 +
  target/linux/ipq807x/config-4.4               |   828 +
  .../arm/boot/dts/qcom-ipq6018-cig-wf188.dts   |    18 +
  .../arm/boot/dts/qcom-ipq6018-cig-wf188n.dts  |    18 +
@@ -66,7 +66,7 @@ Signed-off-by: John Crispin <john@phrozen.org>
  target/linux/ipq807x/patches/112-pstore.patch |   147 +
  .../ipq807x/patches/200-bpf_backport.patch    | 44780 ++++++++++++++++
  toolchain/kernel-headers/Makefile             |     8 +
- 61 files changed, 56686 insertions(+), 2 deletions(-)
+ 61 files changed, 56687 insertions(+), 2 deletions(-)
  create mode 100644 package/boot/uboot-envtools/files/ipq807x
  create mode 100644 target/linux/ipq807x/109-logspam.patch
  create mode 100644 target/linux/ipq807x/Makefile
@@ -4780,10 +4780,10 @@ HcmV?d00001
 
 diff --git a/target/linux/ipq807x/base-files/lib/upgrade/platform.sh b/target/linux/ipq807x/base-files/lib/upgrade/platform.sh
 new file mode 100755
-index 0000000000..407db099f6
+index 0000000000..2c6b682178
 --- /dev/null
 +++ b/target/linux/ipq807x/base-files/lib/upgrade/platform.sh
-@@ -0,0 +1,72 @@
+@@ -0,0 +1,73 @@
 +. /lib/functions/system.sh
 +
 +qca_do_upgrade() {
@@ -4852,6 +4852,7 @@ index 0000000000..407db099f6
 +	edgecore,eap102|\
 +	edgecore,eap101)
 +		CI_UBIPART="rootfs1"
++		[ "$(find_mtd_chardev rootfs)" ] && CI_UBIPART="rootfs"
 +		nand_upgrade_tar "$1"
 +		;;
 +	esac