Update bench_test.go