Fix restore prompt behavior