diff --git a/binaries/fpgadiag/setup.py b/binaries/fpgadiag/setup.py
index de67920978bf..40267ace56a8 100644
--- a/binaries/fpgadiag/setup.py
+++ b/binaries/fpgadiag/setup.py
@@ -24,9 +24,8 @@
 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,  EVEN IF ADVISED OF THE
 # POSSIBILITY OF SUCH DAMAGE.
 
-from setuptools import setup, find_namespace_packages
+from setuptools import Extension, setup, find_namespace_packages
 from setuptools.command.build_ext import build_ext
-from distutils.extension import Extension
 
 # get the original build_extensions method
 original_build_extensions = build_ext.build_extensions
diff --git a/binaries/opae.io/setup.py b/binaries/opae.io/setup.py
index a72621e7fc93..52561a8c15d6 100644
--- a/binaries/opae.io/setup.py
+++ b/binaries/opae.io/setup.py
@@ -24,8 +24,7 @@
 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,  EVEN IF ADVISED OF THE
 # POSSIBILITY OF SUCH DAMAGE.
 
-from setuptools import find_namespace_packages
-from distutils.core import Extension, setup
+from setuptools import Extension, setup, find_namespace_packages
 
 setup(
     name='opae.io',
diff --git a/libraries/pyopae/setup.py b/libraries/pyopae/setup.py
index 3a596f8fa32f..9e84a022f706 100644
--- a/libraries/pyopae/setup.py
+++ b/libraries/pyopae/setup.py
@@ -24,10 +24,8 @@
 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,  EVEN IF ADVISED OF THE
 # POSSIBILITY OF SUCH DAMAGE.
 
-from setuptools import setup, find_namespace_packages
+from setuptools import Extension, setup, find_namespace_packages
 from setuptools.command.build_ext import build_ext
-from distutils.extension import Extension
-
 
 # get the original build_extensions method
 original_build_extensions = build_ext.build_extensions
diff --git a/libraries/pyopaeuio/setup.py b/libraries/pyopaeuio/setup.py
index f2b63b0c3c03..be6cabb1e49a 100644
--- a/libraries/pyopaeuio/setup.py
+++ b/libraries/pyopaeuio/setup.py
@@ -24,9 +24,8 @@
 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,  EVEN IF ADVISED OF THE
 # POSSIBILITY OF SUCH DAMAGE.
 
-from setuptools import setup, find_packages
+from setuptools import Extension, setup, find_packages
 from setuptools.command.build_ext import build_ext
-from distutils.extension import Extension
 
 # get the original build_extensions method
 original_build_extensions = build_ext.build_extensions