minor fix on count