diff --git a/src/pylorax/__init__.py b/src/pylorax/__init__.py
index 2bcc5725..868c482d 100644
--- a/src/pylorax/__init__.py
+++ b/src/pylorax/__init__.py
@@ -48,6 +48,7 @@ import imgutils
 import constants
 from sysutils import *
 
+from treebuilder import TreeBuilder
 from installtree import LoraxInstallTree
 from buildstamp import BuildStamp
 from treeinfo import TreeInfo