- val Set<DataSchemaNode> childNodes = caseNode.childNodes;\r
- if(childNodes !== null) {\r
- resolveDataSchemaNodesAugmented(basePackageName, caseTypeBuilder, childNodes);\r
+ if (caseNode instanceof DataNodeContainer) {\r
+ val DataNodeContainer dataNodeCase = caseNode as DataNodeContainer;\r
+ val Set<DataSchemaNode> childNodes = dataNodeCase.childNodes;\r
+ if(childNodes !== null) {\r
+ resolveDataSchemaNodesAugmented(basePackageName, caseTypeBuilder, childNodes);\r
+ }\r
+ } else {\r
+ val ChoiceCaseNode node = targetNode.getCaseNodeByName(caseNode.getQName().getLocalName());\r
+ val Set<DataSchemaNode> childNodes = node.childNodes;\r
+ if(childNodes !== null) {\r
+ resolveDataSchemaNodesAugmented(basePackageName, caseTypeBuilder, childNodes);\r
+ }\r