[Post] Modify restore procedure