Fix restore which I obviously just broke...