Restore broken websocket support