Get WebSocket test working with API changes.