diff --git a/abl/reasoning/__init__.py b/abl/reasoning/__init__.py
index 8930758..f2ff627 100644
--- a/abl/reasoning/__init__.py
+++ b/abl/reasoning/__init__.py
@@ -1,2 +1,2 @@
-from .reasoner import ReasonerBase
-from .kb import KBBase, prolog_KB
\ No newline at end of file
+from .kb import KBBase, GroundKB, PrologKB
+from .reasoner import ReasonerBase
\ No newline at end of file