fix entity highlighter test