Add flag for auto restore