fix the code highlight