Fixed code highlight