Bug fix re: word count input