Fix the comment counting