diff --git a/setup.py b/setup.py
index 9a38aa6f6188b31cf261a096dbe72e5b119f194b..e38ae79ad5eb6bf6b0d42ae1a887fb356a154d9e 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,7 @@ if sys.version_info < (3, 6, 0):
 
 from gidocgen.core import version
 
-from distutils.command.build_py import build_py as _build_py
+from setuptools.command.build_py import build_py as _build_py
 from setuptools import setup