diff --git a/mmskeleton/datasets/coco.py b/mmskeleton/datasets/coco.py
index 147b4a6eb..c82fb5986 100644
--- a/mmskeleton/datasets/coco.py
+++ b/mmskeleton/datasets/coco.py
@@ -18,7 +18,8 @@
 from .estimation import EstiamtionDataset
 from ..ops.nms.nms import oks_nms
 from ..ops.nms.nms import soft_oks_nms
-from pycocotools import COCO, COCOeval
+from pycocotools.coco import COCO
+from pycocotools.cocoeval import COCOeval
 
 logger = logging.getLogger(__name__)
 
@@ -420,4 +421,4 @@ def _do_python_keypoint_eval(self, res_file, res_folder):
         for ind, name in enumerate(stats_names):
             info_str.append((name, coco_eval.stats[ind]))
 
-        return info_str
\ No newline at end of file
+        return info_str