[BugFix] Fix code highlight bug.