1 files changed, 4 insertions, 1 deletions
diff --git a/share/hydractl/upgrade b/share/hydractl/upgrade
index d0ccde9..5a2f597 100755
--- a/share/hydractl/upgrade
+++ b/share/hydractl/upgrade
@@ -16,6 +16,9 @@
 # License along with this program.  If not, see
 # <http://www.gnu.org/licenses/>.
 
+# Basic parameters
+CLEAN="$1"
+
 # Set sudo config
 if [ "`whoami`" != 'root' ]; then
   sudo="sudo"
@@ -33,6 +36,6 @@ fi
 hydractl chroot-upgrade
 
 # Cleanup
-if [ "$2" == "clean" ]; then
+if [ "$CLEAN" == "clean" ]; then
   $sudo aptitude clean
 fi