Fix to benchmark