Correctly highlight comments.