Fixup Augmentable and Identifiable methods changing
[netvirt.git] / elanmanager / impl / src / main / java / org / opendaylight / netvirt / elan / evpn / listeners / EvpnElanInstanceListener.java
index cdf478f52fc57feb436253e81ffd5ce7e485a572..745d93ff318379d1ab2b5216ff9d0a663ec0d5fc 100644 (file)
@@ -69,13 +69,13 @@ public class EvpnElanInstanceListener extends AsyncDataTreeChangeListenerBase<El
                           ElanInstance update) {
         String elanName = update.getElanInstanceName();
         if (evpnUtils.isWithdrawEvpnRT2Routes(original, update)) {
-            evpnUtils.withdrawEvpnRT2Routes(original.getAugmentation(EvpnAugmentation.class), elanName);
+            evpnUtils.withdrawEvpnRT2Routes(original.augmentation(EvpnAugmentation.class), elanName);
             evpnMacVrfUtils.updateEvpnDmacFlows(original, false);
             evpnUtils.programEvpnL2vniDemuxTable(elanName,
                 (elan, interfaceName) -> evpnUtils.bindElanServiceToExternalTunnel(elanName, interfaceName),
                     mdsalManager::installFlow);
         } else if (evpnUtils.isAdvertiseEvpnRT2Routes(original, update)) {
-            evpnUtils.advertiseEvpnRT2Routes(update.getAugmentation(EvpnAugmentation.class), elanName);
+            evpnUtils.advertiseEvpnRT2Routes(update.augmentation(EvpnAugmentation.class), elanName);
             evpnMacVrfUtils.updateEvpnDmacFlows(update, true);
             evpnUtils.programEvpnL2vniDemuxTable(elanName,
                 (elan, interfaceName) -> evpnUtils.unbindElanServiceFromExternalTunnel(elanName, interfaceName),