diff --git a/scripts/pro/install.sh b/scripts/pro/install.sh
index 6a388dc190..781ecd8caa 100755
--- a/scripts/pro/install.sh
+++ b/scripts/pro/install.sh
@@ -18,6 +18,10 @@ git clone https://$PERSONAL_ACCESS_TOKEN@github.com/Budibase/budibase-pro.git
 if [[ -d "budibase-pro" ]]; then
   cd budibase-pro
 
+  if [[ -z "${BRANCH}" ]]; then
+    export BRANCH=$GITHUB_REF_NAME
+  fi
+
   # Try to checkout the matching pro branch
   git checkout $BRANCH