diff --git a/setup.py b/setup.py
index 3e7ef31d4e160e627ae7abc3d5e93b5395eca75c..12172c952a12e463c35fd2d5ff6aec7d68c39363 100644
--- a/setup.py
+++ b/setup.py
@@ -26,7 +26,10 @@ setuptools.setup(
         "Topic :: Scientific/Engineering :: Astronomy"
     ],
     package_dir={'csst_cpic_sim': 'csst_cpic_sim'},
-    include_package_data=False,
+    include_package_data=True,
+    package_data={
+        'csst_cpic_sim': ['data/*.yaml']
+    },
 
     install_requires=requirements,
     python_requires='>=3.8',