diff --git a/setup.py b/setup.py
index a7be6bf..28c3375 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,10 @@
-from distutils.core import setup
+import sys
+
+if sys.version_info < (3, 7):
+    from distutils.core import setup
+else:
+    from setuptools import setup
+
 setup(
     name='python-logstash',
     packages=['logstash'],