diff --git a/qim3d/viz/_cc.py b/qim3d/viz/_cc.py
index 4992b73402b42914ec3d24e413fe24ecad7c3159..24d047afbb9442f1a05d847dbe836ae84058f520 100644
--- a/qim3d/viz/_cc.py
+++ b/qim3d/viz/_cc.py
@@ -2,7 +2,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 import qim3d
 from qim3d.utils._logger import log
-from segmentation._connected_components import CC
+from qim3d.segmentation._connected_components import CC
 
 def plot_cc(
     connected_components: CC,