Conform -U option to set action and not immediately execute

1 files changed, 5 insertions, 2 deletions

diff --git a/squashfu b/squashfu
index 6c43084..c6e211e 100755
--- a/squashfu
+++ b/squashfu
@@ -352,6 +352,10 @@ action_resquash_now () {
     exit $?
 }
 
+action_unmount () {
+    unmount_all
+}
+
 usage () {
     info "SquashFu: Super Awesome Backup Express (Professional Edition)"
     echo "version: $VER"
@@ -400,8 +404,7 @@ while getopts :BCQR:U opt; do
             action="rollback $OPTARG"
             ;;
         U)
-            unmount_all
-            exit 0
+            action=unmount
             ;;
         c)
             [[ -f $OPTARG ]] && source $OPTARG