Fixed bug in menu highlighter

(it's still terrible though)