1 files changed, 3 insertions, 3 deletions
diff --git a/setup.py b/setup.py
index 6357f28..bfe2c66 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,4 @@
-from setuptools import setup
+from setuptools import find_packages, setup
 
 setup(
     name='SnakeGame',
@@ -6,13 +6,13 @@ setup(
     description='The game of Snake, for beginner AI bot writers.',
     author='Peter Ward',
     author_email='peteraward@gmail.com',
-    packages=['snakegame'],
+    packages=find_packages(),
     zip_safe=False,
     install_requires=[
         'six',
     ],
     package_data={
-        'snakegame': 'images/*.png',
+        'snakegame': ['images/*.png'],
     },
     entry_points={
         'console_scripts': [