diff --git a/install.sh b/install.sh
index b5c9c677fc8f241e69b68dfcadab5b781f58fa7e..b91f8bcae0bd6054678ef6d74cb21368f8da0492 100755
--- a/install.sh
+++ b/install.sh
@@ -2,12 +2,11 @@
 mode="dev"
 
 
-if [ -n "$2" ]; then
-	if [[ $1 =~ ^(prod)$ ]]; then 
-  		mode=$1 
-	fi
+if [[ $* =~ ^(prod)$ ]]; then 
+    mode=$1 
 fi
 
+
 if [ $mode == "dev" ]; then
     echo "[Installing AURA ENGINE for Development]"
 fi