- final BindingDOMDataTreeCommitCohortAdapter<D> adapter =
- new BindingDOMDataTreeCommitCohortAdapter<>(codec, cohort);
- final DOMDataTreeIdentifier domPath = codec.toDOMDataTreeIdentifier(subtree);
- final DOMDataTreeCommitCohortRegistration<?> domReg = registry.registerCommitCohort(domPath, adapter);
- return new ObjectRegistration<T>() {
-
+ final var target = subtree.path().getTargetType();
+ final var adapter = new BindingDOMDataTreeCommitCohortAdapter<>(adapterContext(), cohort,
+ Augmentation.class.isAssignableFrom(target) ? target : null);
+ final var domPath = currentSerializer().toDOMDataTreeIdentifier(subtree);
+ final var domReg = getDelegate().registerCommitCohort(domPath, adapter);
+ return new ObjectRegistration<>() {