fixed a bug with highlighting words