diff --git a/celer/cython_utils.pyx b/celer/cython_utils.pyx
index f56a2f5a..3c297adc 100644
--- a/celer/cython_utils.pyx
+++ b/celer/cython_utils.pyx
@@ -10,8 +10,7 @@ cimport numpy as np
 from scipy.linalg.cython_blas cimport ddot, dasum, daxpy, dnrm2, dcopy, dscal
 from scipy.linalg.cython_blas cimport sdot, sasum, saxpy, snrm2, scopy, sscal
 from scipy.linalg.cython_lapack cimport sposv, dposv
-from libc.math cimport fabs, log, exp, sqrt
-from numpy.math cimport INFINITY
+from libc.math cimport fabs, log, exp, sqrt, INFINITY
 from cython cimport floating
 
 
diff --git a/celer/group_fast.pyx b/celer/group_fast.pyx
index cc671eb1..0e8bab48 100644
--- a/celer/group_fast.pyx
+++ b/celer/group_fast.pyx
@@ -7,9 +7,8 @@ import numpy as np
 cimport numpy as np
 import warnings
 
-from numpy.math cimport INFINITY
 from cython cimport floating
-from libc.math cimport fabs, sqrt
+from libc.math cimport fabs, sqrt, INFINITY
 from sklearn.exceptions import ConvergenceWarning
 
 from .cython_utils cimport (fdot, fasum, faxpy, fnrm2, fcopy, fscal, dual,
diff --git a/celer/multitask_fast.pyx b/celer/multitask_fast.pyx
index 89ff30bb..345d5a10 100644
--- a/celer/multitask_fast.pyx
+++ b/celer/multitask_fast.pyx
@@ -5,8 +5,7 @@ cimport numpy as np
 import numpy as np
 import warnings
 from cython cimport floating
-from libc.math cimport fabs, sqrt
-from numpy.math cimport INFINITY
+from libc.math cimport fabs, sqrt, INFINITY
 from sklearn.exceptions import ConvergenceWarning
 
 from .cython_utils cimport fscal, fcopy, fnrm2, fdot, faxpy