Fix to the augmenting