restore bug fix