Fix code highlight