update restore as well