fix bug in benchmark