fix bug in benchmarking
fix bug in benchmarking