minor fix of benchmark codes