diff --git a/base/setup.py b/base/setup.py
index 76e336e..7686c78 100644
--- a/base/setup.py
+++ b/base/setup.py
@@ -7,12 +7,12 @@
 
 from shutil import copy
 from pathlib import Path
-from distutils.errors import CompileError
 
 try:
     import cython
 except ImportError as e:
     raise ImportError('PyNucleus requires \'Cython\'. Please install it.') from e
+from Cython.Compiler.Errors import CompileError
 
 try:
     import numpy
diff --git a/nl/setup.py b/nl/setup.py
index 2f84c1b..84cd9e3 100644
--- a/nl/setup.py
+++ b/nl/setup.py
@@ -13,6 +13,7 @@
     import cython
 except ImportError as e:
     raise ImportError('PyNucleus requires \'Cython\'. Please install it.') from e
+from Cython.Compiler.Errors import CompileError
 
 try:
     from PyNucleus_packageTools import package