more bug fix in benchmarking