diff --git a/automate.sh b/automate.sh
index f8f26ed..1c6ad3b 100644
--- a/automate.sh
+++ b/automate.sh
@@ -13,6 +13,7 @@ echo "Voulez-vous récupérer les fichiers du dépôt ou effectuer un push sur l
 read answer
 echo "Quelle branche est concernée ? [1 = main, 2 = test]"
 read branch
+
 if [ $branch == 1 ] ; then
 	if [ $answer == 1 ] ; then
 		git checkout main
@@ -30,7 +31,7 @@ if [ $branch == 1 ] ; then
 		sleep 2
 		git push -q origin main
 	fi
-else
+elif [ $branch == 2 ] ; then
 	if [ $answer == 1 ] ; then
 		git checkout test
 		echo "Récupération des fichiers à jour dans la branche test..."
@@ -47,4 +48,6 @@ else
 		sleep 2
 		git push -q origin test
 	fi
+else
+	echo "Choisissez une branche et une option valide."
 fi
\ No newline at end of file