diff --git a/merkle_tree_stream/__init__.py b/merkle_tree_stream/__init__.py
index 4dc634a..00aa539 100644
--- a/merkle_tree_stream/__init__.py
+++ b/merkle_tree_stream/__init__.py
@@ -1,5 +1,10 @@
 """merkle-tree-stream module."""
 
+from merkle_tree_stream.generator import (  # noqa
+    MerkleTreeGenerator,
+    MerkleTreeNode,
+)
+
 try:
     import pkg_resources
 except ImportError: