diff --git a/tests/update_reference_data.sh b/tests/update_reference_data.sh
index f67023f63..37822df23 100755
--- a/tests/update_reference_data.sh
+++ b/tests/update_reference_data.sh
@@ -246,7 +246,7 @@ done
 cd $OPM_TESTS_ROOT
 if [ -n "$BRANCH_NAME" ]
 then
-  git checkout -b $BRANCH_NAME origin/master
+  git checkout -b $BRANCH_NAME $BRANCH_BASE
 fi
 
 # Add potential new files