diff --git a/nanonote-files/script-files/usr/bin/mtd.nn b/nanonote-files/script-files/usr/bin/mtd.nn
index ea41520..f15e1ad 100755
--- a/nanonote-files/script-files/usr/bin/mtd.nn
+++ b/nanonote-files/script-files/usr/bin/mtd.nn
@@ -74,8 +74,8 @@ if [ "$1" == "fw_setenv_default" ]; then
     fw_setenv bootargsf2 mem=32M console=tty0 console=ttyS0,57600n8 rootfstype=ext2 root=/dev/mmcblk0p2 rw rootwait
     fw_setenv bootcmdf3 mmc init\; ext2load mmc 0:3 0x80600000 /boot/uImage\; bootm
     fw_setenv bootargsf3 mem=32M console=tty0 console=ttyS0,57600n8 rootfstype=ext2 root=/dev/mmcblk0p3 rw rootwait
-    fw_setenv bootcmdf4 mmc init\; ext2load mmc 0:4 0x80600000 /boot/uImage\; bootm
-    fw_setenv bootargsf4 mem=32M console=tty0 console=ttyS0,57600n8 rootfstype=ext2 root=/dev/mmcblk0p4 rw rootwait
+    fw_setenv bootcmdf4 mtdparts default\;ubi part rootfs\;ubifsmount rootfs\;ubifsload 0x80600000 /boot/uImage\; bootm
+    fw_setenv bootargsf4 mem=32M console=tty0 console=ttyS0,57600n8 ubi.mtd=2 rootfstype=ubifs root=ubi0:rootfs rw rootwait
     exit 0
 fi