diff --git a/setup.py b/setup.py
index f3870352002bd3acbe981c81c1de49cab8a3f948..db9b26013fb978c8108047d0dc3a5b197b9e05c7 100644
--- a/setup.py
+++ b/setup.py
@@ -39,7 +39,7 @@ def build(dirname):
     packages = find_packages(dirname, include=('fleetrec.*'))
     package_dir = {'': dirname}
     package_data = {}
-    need_copy = ['data/*/*.txt', '*.yaml']
+    need_copy = ['data/*/*.txt', '*.yaml', 'tree/*.npy','tree/*.txt']
     for package in packages:
         if package.startswith("fleetrec.models."):
             package_data[package] = need_copy