diff --git a/entcl/utils/ncd.py b/entcl/utils/ncd.py
index 7093397118f03cb1ef63cc2535b32a26e1319d19..2fe4a5801e2139ffb8ff326e829c648a0cef3894 100644
--- a/entcl/utils/ncd.py
+++ b/entcl/utils/ncd.py
@@ -5,7 +5,7 @@ from loguru import logger
 import numpy as np
 from sklearn.cluster import KMeans
 from sklearn.metrics import confusion_matrix
-from sklearn.utils.linear_assignment_ import linear_sum_assignment
+from scipy.optimize import linear_sum_assignment
 import torch
 from tqdm import tqdm