Improve test for benchmark.