diff --git a/qim3d/utils/connected_components.py b/qim3d/utils/connected_components.py
index 6cccc4896f9b08456da0306e1474d05ac7a1d57c..abb182fae655c53808feb98fb578ab56400f1e82 100644
--- a/qim3d/utils/connected_components.py
+++ b/qim3d/utils/connected_components.py
@@ -88,10 +88,5 @@ def get_3d_connected_components(image: np.ndarray | torch.Tensor):
     Returns:
         class: Returns class object of the connected components.
     """
-    if image.ndim != 3:
-        raise ValueError(
-            f"Given array is not a volume! Current dimension: {image.ndim}"
-        )
-
     connected_components, num_connected_components = label(image)
     return ConnectedComponents(connected_components, num_connected_components)