Update JS to handle websocket update.