Fix code highlight problem