Highlighting fix (minor)