diff --git a/official/vision/keypoints/inference.py b/official/vision/keypoints/inference.py
index 7c1c49b51fb98b3aaa9d55968df299ebfbf18a2a..c3412d23aae7544c0b645dc189af7f2b04c10e21 100644
--- a/official/vision/keypoints/inference.py
+++ b/official/vision/keypoints/inference.py
@@ -20,8 +20,8 @@ from official.vision.keypoints.transforms import get_affine_transform
 from official.vision.keypoints.config import Config as cfg
 
 import official.vision.keypoints.models as M
-import official.vision.detection.retinanet_res50_coco_1x_800size as Det
-from official.vision.detection.tools.test import DetEvaluator
+import official.vision.detection.configs as Det
+from official.vision.detection.tools.utils import DetEvaluator
 from official.vision.keypoints.test import find_keypoints
 
 logger = mge.get_logger(__name__)