diff --git a/abl/reasoning/__init__.py b/abl/reasoning/__init__.py
index 9b3d244..64d3ba4 100644
--- a/abl/reasoning/__init__.py
+++ b/abl/reasoning/__init__.py
@@ -1,2 +1,5 @@
-from .kb import KBBase, ground_KB, prolog_KB
+from .base_kb import BaseKB
+from .ground_kb import GroundKB
+from .prolog_based_kb import PrologBasedKB
 from .reasoner import ReasonerBase
+from .search_based_kb import SearchBasedKB