Add compatibility for websockets