Refactor simple statements

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

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/organization/OrganizationStatementSupport.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/organization/OrganizationStatementSupport.java
index c14e77afc5b8886ab83a3037445cfdf0fa12dce2..b78e2622e5c9e4494f48ddb92ff9c9cb9d57b256 100644 (file)
--- a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/organization/OrganizationStatementSupport.java
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/organization/OrganizationStatementSupport.java
@@ -40,12 +40,12 @@ public final class OrganizationStatementSupport
     @Override
     protected OrganizationStatement createDeclared(final StmtContext<String, OrganizationStatement, ?> ctx,
             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
-        return new RegularOrganizationStatement(ctx, substatements);
+        return new RegularOrganizationStatement(ctx.coerceRawStatementArgument(), substatements);
     }
 
     @Override
     protected OrganizationStatement createEmptyDeclared(final StmtContext<String, OrganizationStatement, ?> ctx) {
-        return new EmptyOrganizationStatement(ctx);
+        return new EmptyOrganizationStatement(ctx.coerceRawStatementArgument());
     }
 
     @Override