diff --git a/missingpy/knnimpute.py b/missingpy/knnimpute.py
index a7f3d1c..ef6379d 100644
--- a/missingpy/knnimpute.py
+++ b/missingpy/knnimpute.py
@@ -10,8 +10,8 @@
 from sklearn.utils import check_array
 from sklearn.utils.validation import check_is_fitted
 from sklearn.utils.validation import FLOAT_DTYPES
-from sklearn.neighbors.base import _check_weights
-from sklearn.neighbors.base import _get_weights
+from sklearn.neighbors._base import _check_weights
+from sklearn.neighbors._base import _get_weights
 
 from .pairwise_external import pairwise_distances
 from .pairwise_external import _get_mask