diff --git a/deployment/setup_18.04.sh b/deployment/setup_18.04.sh
index 8067b6372..152b1feb4 100644
--- a/deployment/setup_18.04.sh
+++ b/deployment/setup_18.04.sh
@@ -48,7 +48,11 @@ rvm use 2.7.2 --default
 
 git clone https://github.com/chatwoot/chatwoot.git
 cd chatwoot
-git checkout master
+if [[ -z $1 ]]; then
+  git checkout master;
+else
+  git checkout $1;
+fi
 bundle
 yarn
 
diff --git a/deployment/setup_20.04.sh b/deployment/setup_20.04.sh
index cc30161c3..effad6402 100644
--- a/deployment/setup_20.04.sh
+++ b/deployment/setup_20.04.sh
@@ -48,7 +48,11 @@ rvm use 2.7.2 --default
 
 git clone https://github.com/chatwoot/chatwoot.git
 cd chatwoot
-git checkout master
+if [[ -z $1 ]]; then
+  git checkout master;
+else
+  git checkout $1;
+fi
 bundle
 yarn