fix .count bug