}
-grub_extn=4
###### begin command line parsing #####
reboot=true
e umount $mount_point
done
-e mount $boot_disk$grub_extn $mount_point
+if [[ $(blockdev --getsize64 ${boot_disk}4) == 8388608 ]]; then
+ # old partition scheme
+ grub_dev=${boot_disk}4
+else
+ grub_dev=${boot_disk}5
+fi
+
+e mount $grub_dev $mount_point
e grub-editenv $mount_point/grubenv set last_boot=/$distro
e grub-editenv $mount_point/grubenv set did_fai_check=true
e umount $mount_point