diff --git a/tests/_openmp/setup.py b/tests/_openmp/setup.py
index 63273d99..aff8d64a 100644
--- a/tests/_openmp/setup.py
+++ b/tests/_openmp/setup.py
@@ -1,8 +1,7 @@
 import os
 import sys
-from distutils.core import setup
+from setuptools import setup, Extension
 from Cython.Build import cythonize
-from distutils.extension import Extension
 
 
 if sys.platform == "darwin":