Forgot a change to support new restore option