CDAP-898 Fix flaky WordCountTest