Highlighting menu - fix bug