change method of restore slightly