Fixing client test