Update example_test.go