Migrate coerceStatementArgument() callers

[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / key / KeyStatementSupport.java

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/key/KeyStatementSupport.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/key/KeyStatementSupport.java
index f6d12de1da17084a2a7bf7f51a812e13280040e1..300e5f71b1ae9770e2111d8baa8709a17071f1d2 100644 (file)
--- a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/key/KeyStatementSupport.java
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/key/KeyStatementSupport.java
@@ -66,7 +66,7 @@ public final class KeyStatementSupport
             final QNameModule targetModule) {
         final Builder<QName> builder = ImmutableSet.builder();
         boolean replaced = false;
-        for (final QName qname : ctx.coerceStatementArgument()) {
+        for (final QName qname : ctx.getArgument()) {
             if (!targetModule.equals(qname.getModule())) {
                 final QName newQname = qname.bindTo(targetModule).intern();
                 builder.add(newQname);
@@ -78,7 +78,7 @@ public final class KeyStatementSupport
 
         // This makes sure we reuse the collection when a grouping is
         // instantiated in the same module
-        return replaced ? builder.build() : ctx.getStatementArgument();
+        return replaced ? builder.build() : ctx.argument();
     }
 
     @Override
@@ -89,12 +89,12 @@ public final class KeyStatementSupport
     @Override
     protected KeyStatement createDeclared(final StmtContext<Set<QName>, KeyStatement, ?> ctx,
             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
-        return new RegularKeyStatement(ctx.getRawArgument(), ctx.coerceStatementArgument(), substatements);
+        return new RegularKeyStatement(ctx.getRawArgument(), ctx.getArgument(), substatements);
     }
 
     @Override
     protected KeyStatement createEmptyDeclared(final StmtContext<Set<QName>, KeyStatement, ?> ctx) {
-        return new EmptyKeyStatement(ctx.getRawArgument(), ctx.coerceStatementArgument());
+        return new EmptyKeyStatement(ctx.getRawArgument(), ctx.getArgument());
     }
 
     @Override