Updated restore process