Fixed issue with WebSockets