diff --git a/hubconf.py b/hubconf.py
index 775092fd75a5a22578a5faa57a0a621190d78bfc..3f0d37d5b354983548679a2f785dd5c5301b7555 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -28,10 +28,10 @@ from official.nlp.bert.model import (
     wwm_cased_L_24_H_1024_A_16,
 )
 
-from official.vision.detection.retinanet_res50_1x_800size import (
-    retinanet_res50_1x_800size,
-    RetinaNet,
+from official.vision.detection.retinanet_res50_coco_1x_800size import (
+    retinanet_res50_coco_1x_800size,
 )
+from official.vision.detection.models import RetinaNet
 from official.vision.detection.tools.test import DetEvaluator
 
 from official.vision.segmentation.deeplabv3plus import (