diff --git a/setup.py b/setup.py
index f2c27bc..dfb15ef 100644
--- a/setup.py
+++ b/setup.py
@@ -32,8 +32,7 @@ setup(
     ],
     python_requires='>=3.6',
     include_package_data=True,
-    data_files=["squirrelbattle/assets/" + file
-                for file in os.listdir("squirrelbattle/assets")],
+    package_data={"squirrelbattle": ["assets/*"]},
     entry_points={
         "console_scripts": [
             "squirrel-battle = squirrelbattle.bootstrap:Bootstrap.run_game",