Fix LeafRefValidatation for refs defined in augmentations