Rewrite augmented assignment in normalization stage