diff --git a/tests/test_tipc/benchmark_train.sh b/tests/test_tipc/benchmark_train.sh
index ef6140aa7918478bb746fe2f75319275323a2c7a..4b7677c72c423d7ce25e9b317b4a29ac919746b9 100644
--- a/tests/test_tipc/benchmark_train.sh
+++ b/tests/test_tipc/benchmark_train.sh
@@ -61,7 +61,7 @@ function get_repo_name(){
     cur_dir=$(pwd)
     IFS="/"
     arr=(${cur_dir})
-    echo ${arr[-1]}
+    echo ${arr[-2]}
 }
 
 FILENAME=$1
@@ -250,4 +250,4 @@ for batch_size in ${batch_size_list[*]}; do
             fi
         done
     done
-done
\ No newline at end of file
+done