fix up restore