diff --git a/setup.py b/setup.py
index 6c88d53..617d841 100644
--- a/setup.py
+++ b/setup.py
@@ -3,9 +3,8 @@
 from pathlib import Path
 
 import requests
-from pybind11.setup_helpers import Pybind11Extension
+from pybind11.setup_helpers import Pybind11Extension, build_ext
 from setuptools import setup
-from setuptools.command.build_ext import build_ext
 
 __version__ = "0.0.4"