1 files changed, 3 insertions, 5 deletions
diff --git a/src/test/java/org/onap/aai/babel/xml/generator/api/AaiModelGeneratorTest.java b/src/test/java/org/onap/aai/babel/xml/generator/api/AaiModelGeneratorTest.java
index db01cb6..27e158f 100644
--- a/src/test/java/org/onap/aai/babel/xml/generator/api/AaiModelGeneratorTest.java
+++ b/src/test/java/org/onap/aai/babel/xml/generator/api/AaiModelGeneratorTest.java
@@ -37,16 +37,14 @@ public class AaiModelGeneratorTest {
     private AaiModelGenerator generator = new AaiModelGenerator();
 
     @Test
-    public void shouldGenerateModelWithInstantiationType() throws XmlArtifactGenerationException, IOException {
+    public void shouldGenerateModelWithCategory() throws XmlArtifactGenerationException, IOException {
         new ArtifactTestUtils().loadWidgetMappings();
         Model model = new Service();
-        model.populateModelIdentificationInformation(Maps.of("instantiationType", "macro"));
+        model.populateModelIdentificationInformation(Maps.of("category", "NST"));
 
 
         String generatedXml = generator.generateModelFor(model);
 
-        assertThat(generatedXml).containsSubsequence("   <model-vers>\n" +
-                "        <model-ver>\n" +
-                "            <orchestration-type>macro</orchestration-type>");
+        assertThat(generatedXml).containsSubsequence("    <model-role>NST</model-role>");
     }
 }