Update restore script