bug fix

forget a register restore