fix resotre bug