fixed a restore bug