Fixed a bug with the highlighted words