diff --git a/setup.py b/setup.py
index 7ecc4a916d251773233bdb0f79e28a108c38e6cc..5dece3071d48168f985d92e48eff77f0c8a204c6 100644
--- a/setup.py
+++ b/setup.py
@@ -39,8 +39,7 @@ def build(dirname):
 
     packages = find_packages(dirname, include=('fleetrec.*'))
     package_dir = {'': dirname}
-
-    package_data = {'fleetrec.models.rank.dnn': ['data/*/*.txt'], }
+    package_data = {'fleetrec.models.rank.dnn': ['data/*/*.txt', '*.yaml'], }
 
     setup(
         name=about["__title__"],