Fixed a TODO about the word counts