fix benchmark code