Fix count issue on update"