Fixed a count issue