Fix error on update