small bug with count