diff --git a/tools/test_op_benchmark.sh b/tools/test_op_benchmark.sh
index f0937ca7dfa2c5f220ca7d77a000b9b9d29899a9..95e9164bd1b66b9e992212a38cf4e32ec5f9761a 100644
--- a/tools/test_op_benchmark.sh
+++ b/tools/test_op_benchmark.sh
@@ -187,7 +187,7 @@ function run_op_benchmark_test {
   done
   # install tensorflow for testing accuary
   pip install tensorflow==2.3.0 tensorflow-probability
-  for branch_name in "develop" "test_pr"
+  for branch_name in "develop" "test"
   do
     git checkout $branch_name
     [ $? -ne 0 ] && LOG "[FATAL] Missing branch ${branch_name}." && exit 7