Switch to refactored WebSocketContainer