fix bugs of WordCount