bug fix! restore