diff --git a/annotator/mmpkg/mmseg/models/decode_heads/cc_head.py b/annotator/mmpkg/mmseg/models/decode_heads/cc_head.py
index 19a77ac..1f4f5b0 100644
--- a/annotator/mmpkg/mmseg/models/decode_heads/cc_head.py
+++ b/annotator/mmpkg/mmseg/models/decode_heads/cc_head.py
@@ -4,7 +4,10 @@ from ..builder import HEADS
 from .fcn_head import FCNHead
 
 try:
-    from annotator.mmpkg.mmcv.ops import CrissCrossAttention
+    try: 
+        from mmcv.ops import CrissCrossAttention
+    except ImportError:
+        from annotator.mmpkg.mmcv.ops import CrissCrossAttention
 except ModuleNotFoundError:
     CrissCrossAttention = None
 
diff --git a/annotator/mmpkg/mmseg/models/decode_heads/point_head.py b/annotator/mmpkg/mmseg/models/decode_heads/point_head.py
index 487c208..c678276 100644
--- a/annotator/mmpkg/mmseg/models/decode_heads/point_head.py
+++ b/annotator/mmpkg/mmseg/models/decode_heads/point_head.py
@@ -2,8 +2,13 @@
 
 import torch
 import torch.nn as nn
-from annotator.mmpkg.mmcv.cnn import ConvModule, normal_init
-from annotator.mmpkg.mmcv.ops import point_sample
+
+try: 
+    from mmcv.cnn import ConvModule, normal_init
+    from mmcv.ops import point_sample
+except ImportError:
+    from annotator.mmpkg.mmcv.cnn import ConvModule, normal_init
+    from annotator.mmpkg.mmcv.ops import point_sample
 
 from annotator.mmpkg.mmseg.models.builder import HEADS
 from annotator.mmpkg.mmseg.ops import resize
diff --git a/annotator/mmpkg/mmseg/models/decode_heads/psa_head.py b/annotator/mmpkg/mmseg/models/decode_heads/psa_head.py
index 3e74510..ba6fe3a 100644
--- a/annotator/mmpkg/mmseg/models/decode_heads/psa_head.py
+++ b/annotator/mmpkg/mmseg/models/decode_heads/psa_head.py
@@ -8,7 +8,10 @@ from ..builder import HEADS
 from .decode_head import BaseDecodeHead
 
 try:
-    from annotator.mmpkg.mmcv.ops import PSAMask
+    try: 
+        from mmcv.ops import PSAMask
+    except ImportError:
+        from annotator.mmpkg.mmcv.ops import PSAMask
 except ModuleNotFoundError:
     PSAMask = None