- if (delegate instanceof DOMDataTreeChangeService) {
- return ((DOMDataTreeChangeService)delegate).registerDataTreeChangeListener(treeId, listener);
+ final DOMDataTreeChangeService treeService =
+ (DOMDataTreeChangeService) delegate.getSupportedExtensions().get(DOMDataTreeChangeService.class);
+ if (treeService != null) {
+ return treeService.registerDataTreeChangeListener(treeId, listener);