diff --git a/zshot/__init__.py b/zshot/__init__.py
index 2bc0b33..845c7c2 100644
--- a/zshot/__init__.py
+++ b/zshot/__init__.py
@@ -1,4 +1,5 @@
-from zshot.zshot import MentionsExtractor, Linker, Zshot, PipelineConfig, RelationsExtractor  # noqa: F401
+from zshot.zshot import MentionsExtractor, Linker, Zshot, PipelineConfig, \
+    RelationsExtractor, KnowledgeExtractor  # noqa: F401
 from zshot.utils.displacy.displacy import displacy  # noqa: F401
 
 __version__ = '0.0.8'