Update client_test.go