fix word highlighting