diff --git a/src/nemos/typing.py b/src/nemos/typing.py
index 42314b90..7f88be21 100644
--- a/src/nemos/typing.py
+++ b/src/nemos/typing.py
@@ -6,7 +6,7 @@
 import jaxopt
 import pynapple as nap
 from jax.typing import ArrayLike
-from statsmodels.tools.typing import NDArray
+from numpy.typing import NDArray
 
 from .pytrees import FeaturePytree