Fix highlighting of mentions