diff --git a/configs/baselines/e2e_faster_rcnn_X-101-64x4d-FPN_1x.yaml b/configs/baselines/e2e_faster_rcnn_X-101-64x4d-FPN_1x.yaml
index 335949af..0ab1b821 100644
--- a/configs/baselines/e2e_faster_rcnn_X-101-64x4d-FPN_1x.yaml
+++ b/configs/baselines/e2e_faster_rcnn_X-101-64x4d-FPN_1x.yaml
@@ -16,7 +16,7 @@ FPN:
   MULTILEVEL_ROIS: True
   MULTILEVEL_RPN: True
 RESNETS:
-IMAGENET_PRETRAINED_WEIGHTS: 'data/pretrained_model/X-101-64x4d.pkl'
+  IMAGENET_PRETRAINED_WEIGHTS: 'data/pretrained_model/X-101-64x4d.pkl'
   STRIDE_1X1: False  # default True for MSRA; False for C2 or Torch models
   TRANS_FUNC: bottleneck_transformation
   NUM_GROUPS: 64
diff --git a/configs/baselines/e2e_keypoint_rcnn_R-101-FPN_1x.yaml b/configs/baselines/e2e_keypoint_rcnn_R-101-FPN_1x.yaml
index a1f841e2..0fcf1690 100644
--- a/configs/baselines/e2e_keypoint_rcnn_R-101-FPN_1x.yaml
+++ b/configs/baselines/e2e_keypoint_rcnn_R-101-FPN_1x.yaml
@@ -18,7 +18,7 @@ FPN:
   MULTILEVEL_ROIS: True
   MULTILEVEL_RPN: True
 FAST_RCNN:
-  ROI_BOX_HEAD: head_builder.roi_2mlp_head
+  ROI_BOX_HEAD: fast_rcnn_heads.roi_2mlp_head
   ROI_XFORM_METHOD: RoIAlign
   ROI_XFORM_RESOLUTION: 7
   ROI_XFORM_SAMPLING_RATIO: 2
diff --git a/configs/baselines/e2e_keypoint_rcnn_R-101-FPN_s1x.yaml b/configs/baselines/e2e_keypoint_rcnn_R-101-FPN_s1x.yaml
index c8184d91..46613688 100644
--- a/configs/baselines/e2e_keypoint_rcnn_R-101-FPN_s1x.yaml
+++ b/configs/baselines/e2e_keypoint_rcnn_R-101-FPN_s1x.yaml
@@ -18,7 +18,7 @@ FPN:
   MULTILEVEL_ROIS: True
   MULTILEVEL_RPN: True
 FAST_RCNN:
-  ROI_BOX_HEAD: head_builder.roi_2mlp_head
+  ROI_BOX_HEAD: fast_rcnn_heads.roi_2mlp_head
   ROI_XFORM_METHOD: RoIAlign
   ROI_XFORM_RESOLUTION: 7
   ROI_XFORM_SAMPLING_RATIO: 2
diff --git a/configs/baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_1x.yaml b/configs/baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_1x.yaml
index 741c5b33..a28a9879 100644
--- a/configs/baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_1x.yaml
+++ b/configs/baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_1x.yaml
@@ -22,7 +22,7 @@ RESNETS:
   NUM_GROUPS: 32
   WIDTH_PER_GROUP: 8
 FAST_RCNN:
-  ROI_BOX_HEAD: head_builder.roi_2mlp_head
+  ROI_BOX_HEAD: fast_rcnn_heads.roi_2mlp_head
   ROI_XFORM_METHOD: RoIAlign
   ROI_XFORM_RESOLUTION: 7
   ROI_XFORM_SAMPLING_RATIO: 2
diff --git a/configs/baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml b/configs/baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml
index 4e35bb7b..92b42285 100644
--- a/configs/baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml
+++ b/configs/baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml
@@ -22,7 +22,7 @@ RESNETS:
   NUM_GROUPS: 32
   WIDTH_PER_GROUP: 8
 FAST_RCNN:
-  ROI_BOX_HEAD: head_builder.roi_2mlp_head
+  ROI_BOX_HEAD: fast_rcnn_heads.roi_2mlp_head
   ROI_XFORM_METHOD: RoIAlign
   ROI_XFORM_RESOLUTION: 7
   ROI_XFORM_SAMPLING_RATIO: 2
diff --git a/configs/baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_1x.yaml b/configs/baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_1x.yaml
index 47c0d8d0..ae212b12 100644
--- a/configs/baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_1x.yaml
+++ b/configs/baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_1x.yaml
@@ -22,7 +22,7 @@ RESNETS:
   NUM_GROUPS: 64
   WIDTH_PER_GROUP: 4
 FAST_RCNN:
-  ROI_BOX_HEAD: head_builder.roi_2mlp_head
+  ROI_BOX_HEAD: fast_rcnn_heads.roi_2mlp_head
   ROI_XFORM_METHOD: RoIAlign
   ROI_XFORM_RESOLUTION: 7
   ROI_XFORM_SAMPLING_RATIO: 2
diff --git a/configs/baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml b/configs/baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml
index d3d6c5d3..51d5efe3 100644
--- a/configs/baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml
+++ b/configs/baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml
@@ -22,7 +22,7 @@ RESNETS:
   NUM_GROUPS: 64
   WIDTH_PER_GROUP: 4
 FAST_RCNN:
-  ROI_BOX_HEAD: head_builder.roi_2mlp_head
+  ROI_BOX_HEAD: fast_rcnn_heads.roi_2mlp_head
   ROI_XFORM_METHOD: RoIAlign
   ROI_XFORM_RESOLUTION: 7
   ROI_XFORM_SAMPLING_RATIO: 2