Fix websocket message handling on 0.3