fixed bug with highlighting