Implementing the API for Drools 6

kiegroup · Dec 3, 2012 · c598969 · c598969
1 parent 7eb3ecd
commit c598969
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/drools-compiler/src/main/java/org/drools/builder/impl/KnowledgeContainerImpl.java b/drools-compiler/src/main/java/org/drools/builder/impl/KnowledgeContainerImpl.java
@@ -1,6 +1,7 @@
 package org.drools.builder.impl;
 
 import org.drools.compiler.PackageBuilderConfiguration;
+import org.drools.kproject.models.KieBaseModelImpl;
 import org.kie.builder.KieBaseModel;
 import org.kie.builder.KieModuleModel;
 import org.kie.builder.KieSessionModel;
@@ -126,7 +127,7 @@ public List<KBaseUnit> getKBaseUnits(File rootFolder, File sourceFolder) {
     }
 
     public KBaseUnit getKBaseUnit(String kBaseName) {
-        KieBaseModel kieBaseModel = kBases.get(kBaseName);
+        KieBaseModelImpl kieBaseModel = ( KieBaseModelImpl ) kBases.get(kBaseName);
         if (kieBaseModel == null) {
             throw new RuntimeException("Unknown KnowledgeBase: " + kBaseName);
         }