Handle zero augmentation case correctly