1 files changed, 8 insertions, 5 deletions
diff --git a/setup.py b/setup.py
index 4fd24a3..c6c80c7 100755
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,8 @@
 #!/usr/bin/env python3
 
-import distutils
-import DistUtilsExtra.auto
+import distutils.core
+import distutils.command.install
+from DistUtilsExtra.command import build_extra, build_i18n
 import os
 
 
@@ -12,18 +13,20 @@ class install_pylock(distutils.command.install.install):
 
         distutils.command.install.install.run(self)
 
-DistUtilsExtra.auto.setup(
+distutils.core.setup(
     name = 'pylock',
     version = '1',
 
-    data_files = [],
-
     packages = ['pylock'],
     package_data = {
         'pylock': ['data/unlock.ui', 'data/bg.svg'],
     },
 
+    scripts = ['scripts/pylock'],
+
     cmdclass = {
+        'build': build_extra.build_extra,
+        'build_i18n': build_i18n.build_i18n,
         'install': install_pylock,
     },
 )