Fixed restore bug