Fix mocked client test