diff --git a/python/setup.py b/python/setup.py
index b21d787..e074a22 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -1,7 +1,7 @@
 import os
 
-from distutils.core import Extension
-from distutils.core import setup
+from setuptools import Extension
+from setuptools import setup
 
 
 # =====