bug fix

forgot a register restoration