bug fix on update