2 files changed, 3 insertions, 4 deletions
diff --git a/hydra b/hydra
index da78bc6..51c1468 100755
--- a/hydra
+++ b/hydra
@@ -57,7 +57,10 @@ export NAME="$BASENAME"
 # Adjust hydra parameter
 if [ "$BASENAME" == "hydra" ]; then
   export HYDRA="$1"
+  export ACTION="$2"
   shift
+else
+  export ACTION="$1"
 fi
 
 # Load
diff --git a/lib/hydra/misc b/lib/hydra/misc
index a3156a7..f086fb7 100644
--- a/lib/hydra/misc
+++ b/lib/hydra/misc
@@ -7,10 +7,6 @@ function hydra_set_env {
   export COMMIT="`( cd $APP_BASE && git log -n 1 --pretty=oneline 2> /dev/null | cut -d " " -f 1 )`"
   export CONFIG="$HOME/.hydra/config"
 
-  if [ -z "$ACTION" ]; then
-    export ACTION="$1"
-  fi
-
   if [ "`whoami`" != 'root' ]; then
     SUDO="sudo"
   fi