fixed restore bug