Fix diff highlight