Fix problem about restore steps