Update behavior of count