Fix restore bug

Going to the corner to put on my dunce hat.