3 files changed, 5 insertions, 6 deletions
diff --git a/src/main/java/org/onap/aai/datagrooming/DataGrooming.java b/src/main/java/org/onap/aai/datagrooming/DataGrooming.java
index e222228..7e667f4 100644
--- a/src/main/java/org/onap/aai/datagrooming/DataGrooming.java
+++ b/src/main/java/org/onap/aai/datagrooming/DataGrooming.java
@@ -316,7 +316,7 @@ public class DataGrooming {
 			throw aai;
 		}
 		LoaderFactory loaderFactory = ctx.getBean(LoaderFactory.class);
-		SchemaVersions schemaVersions = ctx.getBean(SchemaVersions.class);
+		SchemaVersions schemaVersions = (SchemaVersions) ctx.getBean("schemaVersions");
 		DataGrooming dataGrooming = new DataGrooming(loaderFactory, schemaVersions);
 		dataGrooming.execute(args);
 	}
diff --git a/src/main/java/org/onap/aai/db/schema/ScriptDriver.java b/src/main/java/org/onap/aai/db/schema/ScriptDriver.java
index c45ad28..6b36adc 100644
--- a/src/main/java/org/onap/aai/db/schema/ScriptDriver.java
+++ b/src/main/java/org/onap/aai/db/schema/ScriptDriver.java
@@ -90,7 +90,7 @@ public class ScriptDriver {
 			throw aai;
 		}
 		AuditorFactory auditorFactory = ctx.getBean(AuditorFactory.class);
-		SchemaVersions schemaVersions = ctx.getBean(SchemaVersions.class);
+		SchemaVersions schemaVersions = (SchemaVersions) ctx.getBean("schemaVersions");
 		EdgeIngestor edgeIngestor     = ctx.getBean(EdgeIngestor.class);
 
 		String config = cArgs.config;
@@ -136,4 +136,4 @@ class CommandLineArgs {
 	public String type = "graph";
 	
 
-}
-\ No newline at end of file
+}
diff --git a/src/test/java/org/onap/aai/AAISetup.java b/src/test/java/org/onap/aai/AAISetup.java
index 5ceafa7..25e011b 100644
--- a/src/test/java/org/onap/aai/AAISetup.java
+++ b/src/test/java/org/onap/aai/AAISetup.java
@@ -50,9 +50,8 @@ import org.springframework.test.context.junit4.rules.SpringClassRule;
 import org.springframework.test.context.junit4.rules.SpringMethodRule;
 
 @ContextConfiguration(classes = {
-        SchemaLocationsBean.class,
+        ConfigConfiguration.class,
         AAIConfigTranslator.class,
-        SchemaVersions.class,
         NodeIngestor.class,
         EdgeIngestor.class,
         EdgeSerializer.class,
@@ -135,4 +134,4 @@ public abstract class AAISetup {
         String resource = IOUtils.toString(inputStream);
         return resource;
     }
-}
-\ No newline at end of file
+}