diff --git a/cmd/root.go b/cmd/root.go
index b036f0f..4c9ae4e 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -43,24 +43,27 @@ var rootCmd = &cobra.Command{
 func downgrade() error {
 	fmt.Println("Downgrading ", PkgName, "to ", Version)
 
+	if Mask && Unmask {
+		return errors.New("please choose either --mask or --unmask, not both")
+	}
+
 	cmd := exec.Command("flatpak", "update", "--commit="+Version, PkgName)
 
 	cmd.Stdin = os.Stdin
 	cmd.Stdout = os.Stdout
 	cmd.Stderr = os.Stderr
 
+	if Unmask {
+		unmask()
+	}
+
 	err := cmd.Run()
 	if err != nil {
 		return err
 	}
 
-	if Mask && Unmask {
-		return errors.New("please choose either --mask or --unmask, not both")
-	}
 	if Mask {
 		mask()
-	} else if Unmask {
-		unmask()
 	}
 
 	return nil