diff --git a/dwi_ml/data/processing/streamlines/post_processing.py b/dwi_ml/data/processing/streamlines/post_processing.py
index 21356b83..28b15dff 100644
--- a/dwi_ml/data/processing/streamlines/post_processing.py
+++ b/dwi_ml/data/processing/streamlines/post_processing.py
@@ -8,7 +8,7 @@
 from matplotlib.colors import LogNorm
 from mpl_toolkits.axes_grid1 import make_axes_locatable
 
-from scilpy.tractanalysis.tools import \
+from scilpy.tractanalysis.connectivity_segmentation import \
     extract_longest_segments_from_profile as segmenting_func
 from scilpy.tractograms.uncompress import uncompress