Fixed issue with code highlight