diff --git a/scripts/addon.d.sh b/scripts/addon.d.sh
index f1200b3a2..c8ec4127e 100644
--- a/scripts/addon.d.sh
+++ b/scripts/addon.d.sh
@@ -43,12 +43,12 @@ show_logo() {
 
 installation() {
   find_manager_apk
+  get_flags
   find_boot_image
   find_dtbo_image
   [ -z $BOOTIMAGE ] && abort "! Unable to detect target image"
   ui_print "- Target image: $BOOTIMAGE"
   [ -z $DTBOIMAGE ] || ui_print "- DTBO image: $DTBOIMAGE"
-  get_flags
 
   remove_system_su
 
diff --git a/scripts/flash_script.sh b/scripts/flash_script.sh
index c8d26795d..df14ef3bf 100644
--- a/scripts/flash_script.sh
+++ b/scripts/flash_script.sh
@@ -41,12 +41,10 @@ ui_print "************************"
 ui_print "* Magisk v$MAGISK_VER Installer"
 ui_print "************************"
 
-is_mounted /data || mount /data || is_mounted /cache || mount /cache || abort "! Unable to mount partitions"
+is_mounted /data || mount /data || is_mounted /cache || mount /cache
 mount_partitions
-
 check_data
 get_flags
-
 find_boot_image
 
 [ -z $BOOTIMAGE ] && abort "! Unable to detect target image"