Fix word count bug