diff --git a/sam2_realtime/utils/transforms.py b/sam2_realtime/utils/transforms.py
index d05cd3e..88ec3e5 100644
--- a/sam2_realtime/utils/transforms.py
+++ b/sam2_realtime/utils/transforms.py
@@ -75,7 +75,7 @@ class SAM2Transforms(nn.Module):
         """
         Perform PostProcessing on output masks.
         """
-        from sam2.utils.misc import get_connected_components
+        from sam2_realtime.utils.misc import get_connected_components
 
         masks = masks.float()
         if self.max_hole_area > 0: