diff --git a/backwards-compat-tests/scripts/run_local.sh b/backwards-compat-tests/scripts/run_local.sh
index cfa164ec35a29..bf2f09c6841a4 100755
--- a/backwards-compat-tests/scripts/run_local.sh
+++ b/backwards-compat-tests/scripts/run_local.sh
@@ -46,13 +46,14 @@ EOF
 
 setup_old_cluster() {
   echo "--- Setting up old cluster"
+  LATEST_BRANCH=$(git branch --show-current)
   git checkout "v${OLD_VERSION}-rc"
 }
 
 setup_new_cluster() {
   echo "--- Setting up new cluster"
   rm -r .risingwave/bin/risingwave
-  git checkout main
+  git checkout $LATEST_BRANCH
 }
 
 main() {