diff --git a/install_yunohostv2 b/install_yunohostv2
index 68bcff3..afadf48 100755
--- a/install_yunohostv2
+++ b/install_yunohostv2
@@ -98,9 +98,10 @@ then
 		echo "deb http://repo.yunohost.org/ lemonldap main" >> $CUSTOMAPT
 	fi
 	
-	if [[ "$1" == "test" ]]
-	then
+	if [ $# -eq 1 ]; then
+	    if [[ "$1" == "test" ]]; then
 		echo "deb http://repo.yunohost.org/ test main" >> $CUSTOMAPT
+	    fi
 	fi
 
         #Get gpg key