diff --git a/qim3d/viz/_cc.py b/qim3d/viz/_cc.py
index 63731f2fdd420950f460eccde429397053f813fa..51218787556739abe14e1525c5d9034b11153426 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,