X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=yang%2Frfc7952-parser-support%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fyangtools%2Frfc7952%2Fparser%2FAnnotationStatementSupport.java;h=6436e44e0862bf51f18b3b85ff7685b3180cb140;hb=99f08f31a49febbc65ab8efbbea43aabf0de0f6f;hp=14b8783f6e0ea161f1112f2d19700f0076f8f85e;hpb=eb5ea74052360c816b88f33680116e545f01c8a7;p=yangtools.git

diff --git a/yang/rfc7952-parser-support/src/main/java/org/opendaylight/yangtools/rfc7952/parser/AnnotationStatementSupport.java b/yang/rfc7952-parser-support/src/main/java/org/opendaylight/yangtools/rfc7952/parser/AnnotationStatementSupport.java
index 14b8783f6e..6436e44e08 100644
--- a/yang/rfc7952-parser-support/src/main/java/org/opendaylight/yangtools/rfc7952/parser/AnnotationStatementSupport.java
+++ b/yang/rfc7952-parser-support/src/main/java/org/opendaylight/yangtools/rfc7952/parser/AnnotationStatementSupport.java
@@ -63,7 +63,7 @@ public final class AnnotationStatementSupport
             final StmtContext<String, ?, ?> unitsStmt = StmtContextUtils.findFirstEffectiveSubstatement(
                     stmt.caerbannog(), UnitsStatement.class);
             if (unitsStmt != null) {
-                builder.setUnits(unitsStmt.getStatementArgument());
+                builder.setUnits(unitsStmt.argument());
             }
             type = builder.build();
         }
@@ -123,10 +123,9 @@ public final class AnnotationStatementSupport
 
     @Override
     public void onStatementAdded(final Mutable<QName, AnnotationStatement, AnnotationEffectiveStatement> stmt) {
-        final StatementDefinition parentDef = stmt.coerceParentContext().getPublicDefinition();
+        final StatementDefinition parentDef = stmt.coerceParentContext().publicDefinition();
         SourceException.throwIf(YangStmtMapping.MODULE != parentDef && YangStmtMapping.SUBMODULE != parentDef,
-                stmt.getStatementSourceReference(),
-                "Annotations may only be defined at root of either a module or a submodule");
+                stmt.sourceReference(), "Annotations may only be defined at root of either a module or a submodule");
     }
 
     @Override
@@ -137,7 +136,7 @@ public final class AnnotationStatementSupport
     @Override
     protected AnnotationStatement createDeclared(final StmtContext<QName, AnnotationStatement, ?> ctx,
             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
-        return new Declared(ctx.coerceStatementArgument(), substatements);
+        return new Declared(ctx.getArgument(), substatements);
     }
 
     @Override