Fix restore state