Fix the feature of code highlight