fix bug in highlighting