X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=yang%2Fyang-parser-impl%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fyangtools%2Fyang%2Fstmt%2FTestUtils.java;fp=yang%2Fyang-parser-impl%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fyangtools%2Fyang%2Fstmt%2FTestUtils.java;h=8d39a4b678f8d222f39bb21d51d1de54ba8473d7;hb=1cb3b70ea959108d66dd19cbe30c86d78c77e5c9;hp=737a260438357587f5c737de8c2f6a38ccbf50b0;hpb=b084ee60e84a41e854018599d36719b4200b231d;p=yangtools.git

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/TestUtils.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/TestUtils.java
index 737a260438..8d39a4b678 100644
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/TestUtils.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/TestUtils.java
@@ -153,7 +153,7 @@ public final class TestUtils {
                 checkIsAugmenting(child, expected);
             }
         } else if (node instanceof ChoiceSchemaNode) {
-            for (ChoiceCaseNode caseNode : ((ChoiceSchemaNode) node).getCases()) {
+            for (ChoiceCaseNode caseNode : ((ChoiceSchemaNode) node).getCases().values()) {
                 checkIsAugmenting(caseNode, expected);
             }
         }
@@ -176,7 +176,7 @@ public final class TestUtils {
                 checkIsAddedByUses(child, expected);
             }
         } else if (node instanceof ChoiceSchemaNode) {
-            for (ChoiceCaseNode caseNode : ((ChoiceSchemaNode) node).getCases()) {
+            for (ChoiceCaseNode caseNode : ((ChoiceSchemaNode) node).getCases().values()) {
                 checkIsAddedByUses(caseNode, expected);
             }
         }