1 files changed, 2 insertions, 1 deletions
diff --git a/setup.py b/setup.py
index b8c5320..6a57c41 100755
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-from distutils.core import setup
+from setuptools import setup
 
 version = "0.9.8"
 
@@ -13,6 +13,7 @@ setup(name="fastimport",
       license="GNU GPL v2 or later",
       url="https://github.com/jelmer/python-fastimport",
       packages=['fastimport', 'fastimport.tests', 'fastimport.processors'],
+      test_suite="fastimport.tests.test_suite",
       scripts=[
           'bin/fast-import-query',
           'bin/fast-import-filter',