Update ChoiceSchemaNode design
[yangtools.git] / yang / yang-data-util / src / main / java / org / opendaylight / yangtools / yang / data / util / ChoiceNodeContextNode.java
index e21265581907a593d4c9639a288809e67e314653..27bacac5299206fef5670881a455e3062af6d9e1 100644 (file)
@@ -25,7 +25,7 @@ class ChoiceNodeContextNode extends AbstractMixinContextNode<NodeIdentifier> {
         ImmutableMap.Builder<QName, DataSchemaContextNode<?>> byQNameBuilder = ImmutableMap.builder();
         ImmutableMap.Builder<PathArgument, DataSchemaContextNode<?>> byArgBuilder = ImmutableMap.builder();
 
-        for (ChoiceCaseNode caze : schema.getCases()) {
+        for (ChoiceCaseNode caze : schema.getCases().values()) {
             for (DataSchemaNode cazeChild : caze.getChildNodes()) {
                 DataSchemaContextNode<?> childOp = fromDataSchemaNode(cazeChild);
                 byArgBuilder.put(childOp.getIdentifier(), childOp);