diff --git a/automate.sh b/automate.sh
index 2bc1bd1..2d4417b 100644
--- a/automate.sh
+++ b/automate.sh
@@ -6,17 +6,36 @@
 echo "Voulez-vous récupérer les fichiers à jour ou effectuer un push ? 
 [1 = Pull, 2 = Push]"
 read answer
-if [ $answer == 1 ] ; then
-	echo "Récupération des fichiers à jour..."
-	git pull
+echo "Quelle branche est concernée ? [1 = main, 2 = tests]"
+read branch
+if [ $branch == 1 ] ; then
+	if [ $answer == 1 ] ; then
+		echo "Récupération des fichiers à jour, branche main..."
+		git pull origin main
+	else
+		echo "Ajout des fichiers au Gitea..."
+		sleep 2
+		git add .
+		echo "Commit en cours..."
+		sleep 2
+		git commit
+		echo "Push des fichiers au Gitea, branche main..."
+		sleep 2
+		git push origin main
+	fi
 else
-	echo "Ajout des fichiers au Gitea..."
-	sleep 2
-	git add .
-	echo "Commit en cours..."
-	sleep 2
-	git commit
-	echo "Push des fichiers au Gitea..."
-	sleep 2
-	git push
-fi
+	if [ $answer == 1 ] ; then
+		echo "Récupération des fichiers à jour dans la branche tests..."
+		git pull origin tests
+	else
+		echo "Ajout des fichiers au Gitea, branche tests..."
+		sleep 2
+		git add .
+		echo "Commit en cours..."
+		sleep 2
+		git commit
+		echo "Push des fichiers au Gitea, branche tests..."
+		sleep 2
+		git push origin tests
+	fi
+fi
\ No newline at end of file