Fix highlighting in diff