Fix menu highlight