Fix websocket bug

Now WebSocket is working!