diff --git a/pyproject.toml b/pyproject.toml
index 478b5b42..46015312 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -56,5 +56,8 @@ packages = [
     'pymdp.legacy.envs',
 ]
 
+[tool.setuptools.package-data]
+pymdp = ['envs/assets/*']
+
 [tool.black]
 line-length = 120