@SuppressWarnings("unchecked")
@Override
public final <A extends Augmentation<T>> A augmentation(final Class<A> augmentationType) {
- return (A) augmentations.get(CodeHelpers.nonNullValue(augmentationType, "augmentationType"));
+ return (A) augmentations.get(requireNonNull(augmentationType));
}
@Override