Update for websocket 0.9.1