Improve restore