Count update fix