Fix issue with obtaining WebSocketContainer