Fix mock ws client.