diff --git a/README.md b/README.md
index e07faed..9b4cc2a 100644
--- a/README.md
+++ b/README.md
@@ -2,4 +2,4 @@ Dépôt de fichiers personnels provenant de machines virtuelles utilisées en TP
 
 Ce Gitea contient aussi un script d'automatisation du processus de push et de pull, adaptés aux deux branches.
 
-Ce dépôt étant public, libre à vous de prendre/télécharger des fichiers ou de faire des bifurcations, aucun problème.
\ No newline at end of file
+Le dépôt a été passé en privé de manière possiblement permanente, pour garder mon travail exclusif à moi-même.
\ No newline at end of file
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