Minor changes to benchmark in production