Fixing word count issue