fixes problem with highlighting