fix size of Augmentations