Fix a bug on highlighting.