1 files changed, 2 insertions, 1 deletions
diff --git a/src/main/java/org/onap/aai/graphgraph/App.java b/src/main/java/org/onap/aai/graphgraph/App.java
index 58e0d66..910c086 100644
--- a/src/main/java/org/onap/aai/graphgraph/App.java
+++ b/src/main/java/org/onap/aai/graphgraph/App.java
@@ -20,6 +20,7 @@
 package org.onap.aai.graphgraph;
 
 import static org.onap.aai.graphgraph.ModelExporter.exportModel;
+import static org.onap.aai.graphgraph.ModelExporter.writeExportedModel;
 
 import java.io.IOException;
 import java.nio.file.Files;
@@ -74,7 +75,7 @@ public class App{
         edgeIngestor = (EdgeIngestor) context.getBean("edgeIngestor");
 
         if (parser.shoudGenerateUml()){
-            exportModel(parser.getSchemaVersion());
+            writeExportedModel(exportModel(parser.getSchemaVersion()));
             System.exit(0);
         }
     }