Fix diff highlighting