1 files changed, 3 insertions, 2 deletions
diff --git a/plugins/org.eclipse.osee.ats.config.demo/src/org/eclipse/osee/ats/config/demo/config/DemoDatabaseConfig.java b/plugins/org.eclipse.osee.ats.config.demo/src/org/eclipse/osee/ats/config/demo/config/DemoDatabaseConfig.java
index fde93183fc9..00805775e0a 100644
--- a/plugins/org.eclipse.osee.ats.config.demo/src/org/eclipse/osee/ats/config/demo/config/DemoDatabaseConfig.java
+++ b/plugins/org.eclipse.osee.ats.config.demo/src/org/eclipse/osee/ats/config/demo/config/DemoDatabaseConfig.java
@@ -21,7 +21,6 @@ import org.eclipse.osee.framework.core.data.IOseeBranch;
 import org.eclipse.osee.framework.core.enums.CoreArtifactTypes;
 import org.eclipse.osee.framework.core.exception.OseeCoreException;
 import org.eclipse.osee.framework.core.model.Branch;
-import org.eclipse.osee.framework.database.core.OseeInfo;
 import org.eclipse.osee.framework.database.init.IDbInitializationTask;
 import org.eclipse.osee.framework.skynet.core.OseeSystemArtifacts;
 import org.eclipse.osee.framework.skynet.core.artifact.Artifact;
@@ -30,6 +29,7 @@ import org.eclipse.osee.framework.skynet.core.artifact.BranchManager;
 import org.eclipse.osee.framework.skynet.core.utility.Requirements;
 import org.eclipse.osee.framework.ui.skynet.widgets.workflow.WorkItemDefinition.WriteType;
 import org.eclipse.osee.support.test.util.DemoSawBuilds;
+import org.eclipse.osee.support.test.util.TestUtil;
 
 /**
  * Initialization class that will load configuration information for a sample DB.
@@ -41,6 +41,8 @@ public class DemoDatabaseConfig implements IDbInitializationTask {
    @Override
    public void run() throws OseeCoreException {
 
+      TestUtil.setDemoDb(true);
+
       if (AtsUtil.dbInitWorkItemDefs()) {
          new DemoCodeWorkFlowDefinition().config(WriteType.New, null);
          new DemoTestWorkFlowDefinition().config(WriteType.New, null);
@@ -56,7 +58,6 @@ public class DemoDatabaseConfig implements IDbInitializationTask {
       BranchManager.createTopLevelBranch(DemoCISBuilds.CIS_Bld_1);
       populateProgramBranch(DemoCISBuilds.CIS_Bld_1);
 
-      OseeInfo.putValue(OseeInfo.DB_TYPE_KEY, "demo");
    }
 
    private void populateProgramBranch(IOseeBranch branch) throws OseeCoreException {