Fix menu highlighting classnames