improve bench_test.go