fix code highlight bug