diff --git a/lib/xos-config/setup.py b/lib/xos-config/setup.py
index 2e29a40..a3e33fc 100644
--- a/lib/xos-config/setup.py
+++ b/lib/xos-config/setup.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-from distutils.core import setup
+from setuptools import setup
 
 setup(name='XosConfig',
       version='1.0',
@@ -8,7 +8,5 @@
       author='Matteo Scandolo',
       author_email='teo@onlab.us',
       packages=['xosconfig'],
-      data_files=[
-            ('.', ['xosconfig/config-schema.yaml'])
-      ]
-     )
\ No newline at end of file
+      include_package_data=True
+     )