diff --git a/build.sh b/build.sh
index fb92fc666a..998a650181 100644
--- a/build.sh
+++ b/build.sh
@@ -3,7 +3,7 @@
 RED='\033[0;31m'
 NC='\033[0m' # No Color
 
-echo "Updating git repository"
+echo "Updating git repository $1 / $2"
 
 git fetch origin
 git reset --hard origin/master
@@ -15,7 +15,7 @@ fi
 
 echo 'Starting build...'
 
-docker build --build-arg VERSION="$1" --tag appwrite/appwrite:"$1" .
+docker build --build-arg VERSION="$2" --tag appwrite/appwrite:"$1" .
 
 echo 'Pushing build to registry...'