Fix restore bug.