2 files changed, 3 insertions, 3 deletions
diff --git a/acl/acl-impl/src/main/java/io/fd/hc2vpp/acl/util/ace/AceConverter.java b/acl/acl-impl/src/main/java/io/fd/hc2vpp/acl/util/ace/AceConverter.java
index 228fbfc68..db5efb4d6 100644
--- a/acl/acl-impl/src/main/java/io/fd/hc2vpp/acl/util/ace/AceConverter.java
+++ b/acl/acl-impl/src/main/java/io/fd/hc2vpp/acl/util/ace/AceConverter.java
@@ -119,7 +119,7 @@ public interface AceConverter extends MacIpAceDataExtractor, StandardAceDataExtr
 
             final String aceName = macipAclContext.getAceName(aclName, i++, mappingContext);
             ace.setRuleName(aceName);
-            ace.setKey(new AceKey(aceName));
+            ace.withKey(new AceKey(aceName));
 
             aces.add(ace.build());
         }
@@ -144,7 +144,7 @@ public interface AceConverter extends MacIpAceDataExtractor, StandardAceDataExtr
 
             final String aceName = standardAclContext.getAceName(aclName, i++, mappingContext);
             ace.setRuleName(aceName);
-            ace.setKey(new AceKey(aceName));
+            ace.withKey(new AceKey(aceName));
             aces.add(ace.build());
         }
         return aces;
diff --git a/acl/acl-impl/src/main/java/io/fd/hc2vpp/acl/util/acl/AclDataExtractor.java b/acl/acl-impl/src/main/java/io/fd/hc2vpp/acl/util/acl/AclDataExtractor.java
index bae699109..5f75ae306 100644
--- a/acl/acl-impl/src/main/java/io/fd/hc2vpp/acl/util/acl/AclDataExtractor.java
+++ b/acl/acl-impl/src/main/java/io/fd/hc2vpp/acl/util/acl/AclDataExtractor.java
@@ -66,7 +66,7 @@ public interface AclDataExtractor {
      * Convert {@link Acl} tag to byte array in US_ASCII
      */
     default byte[] getAclTag(@Nonnull final Acl acl) {
-        final VppAclAugmentation augmentation = acl.getAugmentation(VppAclAugmentation.class);
+        final VppAclAugmentation augmentation = acl.augmentation(VppAclAugmentation.class);
         if (augmentation != null && augmentation.getTag() != null) {
             return augmentation.getTag().getBytes(StandardCharsets.US_ASCII);
         }