Update client/client_test.go