1 files changed, 5 insertions, 5 deletions

diff --git a/aai-schema-gen/src/main/java/org/onap/aai/schemagen/swagger/GenerateSwagger.java b/aai-schema-gen/src/main/java/org/onap/aai/schemagen/swagger/GenerateSwagger.java
index b467e33..c050063 100644
--- a/aai-schema-gen/src/main/java/org/onap/aai/schemagen/swagger/GenerateSwagger.java
+++ b/aai-schema-gen/src/main/java/org/onap/aai/schemagen/swagger/GenerateSwagger.java
@@ -45,7 +45,7 @@ import java.util.Set;
 import java.util.TreeMap;
 import java.util.stream.Collectors;
 
-import org.onap.aai.setup.SchemaVersions;
+import org.onap.aai.setup.SchemaConfigVersions;
 
 public class GenerateSwagger {
 
@@ -58,16 +58,16 @@ public class GenerateSwagger {
     // used to check to see if program is run from aai-schema-gen
     public static final String DEFAULT_RUN_DIR = "aai-schema-gen";
 
-    public static SchemaVersions schemaVersions;
+    public static SchemaConfigVersions schemaConfigVersions;
 
-    public SchemaVersions getSchemaVersions() {
-        return schemaVersions;
+    public SchemaConfigVersions getSchemaConfigVersions() {
+        return schemaConfigVersions;
     }
 
     public static void main(String[] args) throws IOException, TemplateException {
 
         // SchemaVersions schemaVersions = SpringContextAware.getBean(SchemaVersions.class);
-        String CURRENT_VERSION = schemaVersions.getDefaultVersion().toString();
+        String CURRENT_VERSION = schemaConfigVersions.getDefaultVersion().toString();
         String schemaDir = System.getProperty("aai.schema.dir");
         String versionToGenerate = System.getProperty("aai.generate.version");
         String wikiLink = System.getProperty("aai.wiki.link");