diff --git a/src/gui/views/treemodels/Makefile.am b/src/gui/views/treemodels/Makefile.am
index 60ad19f19..b4049a5b6 100644
--- a/src/gui/views/treemodels/Makefile.am
+++ b/src/gui/views/treemodels/Makefile.am
@@ -7,7 +7,8 @@ pkgdatadir = $(datadir)/@PACKAGE@/gui/views/treemodels
 
 pkgdata_PYTHON = \
 	__init__.py \
-	flatbasemodel.py
+	flatbasemodel.py \
+	treebasemodel.py
 
 pkgpyexecdir = @pkgpyexecdir@/gui/views/treemodels
 pkgpythondir = @pkgpythondir@/gui/views/treemodels