Fix test/mock/ws_client.go so tests work.