diff --git a/flash.sh b/flash.sh
index 8bec239..5a35b56 100755
--- a/flash.sh
+++ b/flash.sh
@@ -45,14 +45,10 @@
     fi
 }
 
-for i in u-boot.imx ${PART_IMAGE} boot_${USERSPACE_ARCH}.img rootfs_${USERSPACE_ARCH}.img; do
+for i in u-boot.imx boot_${USERSPACE_ARCH}.img rootfs_${USERSPACE_ARCH}.img; do
     [[ ! -f ${PRODUCT_OUT}/$i ]] && die "${PRODUCT_OUT}/$i is missing."
 done
 
-# Flash bootloader
-${FASTBOOT_CMD} flash bootloader0 ${PRODUCT_OUT}/u-boot.imx
-${FASTBOOT_CMD} reboot-bootloader
-
 # Figure out which partition map we need based upon fastboot vars
 MMC_SIZE=$(${FASTBOOT_CMD} getvar mmc_size 2>&1 | awk '/mmc_size:/ { print $2 }')
 PART_IMAGE=$(partition_table_image ${MMC_SIZE})
@@ -62,6 +58,10 @@
     exit 1
 fi
 
+# Flash bootloader
+${FASTBOOT_CMD} flash bootloader0 ${PRODUCT_OUT}/u-boot.imx
+${FASTBOOT_CMD} reboot-bootloader
+
 # Flash partition table
 ${FASTBOOT_CMD} flash gpt ${PRODUCT_OUT}/${PART_IMAGE}
 ${FASTBOOT_CMD} reboot-bootloader