diff --git a/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml b/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml
index 081e371e2099aa189a1a4cc784a5f999a69aa59c..c27b4e6c417e4e3597a0733c83307b17bddc28e7 100644
--- a/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml
+++ b/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml
@@ -98,7 +98,6 @@ CascadeHead:
   reg_class_agnostic: False
   stage_loss_weights: [1, 0.5, 0.25]
   loss_normalize_pos: True
-  add_gt_as_proposals: [True, True, True]
 
 
 BBoxAssigner:
diff --git a/ppdet/modeling/heads/roi_extractor.py b/ppdet/modeling/heads/roi_extractor.py
index c7eb4638219f9ba30b61aee59953ee44ba483868..eb9b75b10e1a6fca88b12f27b48a5e51969a52c2 100644
--- a/ppdet/modeling/heads/roi_extractor.py
+++ b/ppdet/modeling/heads/roi_extractor.py
@@ -96,7 +96,7 @@ class RoIAlign(object):
                     self.canonical_size,
                     rois_num=rois_num)
             else:
-                ops.distribute_fpn_proposals(
+                rois_dist, restore_index, rois_num_dist = ops.distribute_fpn_proposals(
                     roi,
                     k_min,
                     k_max,