diff --git a/setup.py b/setup.py
index 5a184a59..81de6eb8 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,8 @@
 from dataclasses import dataclass
 
 from setuptools import setup, Extension, Command
-from setuptools.command import build_ext, sdist, build
+from setuptools.command import build_ext, sdist
+from distutils.command import build
 
 # This is used to find the compiler and flags for building the test extension
 import distutils.ccompiler