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 19ce3df826f06f52f01a34a65117609a35162555..a79a7fd15124a39800e8d8e3aae1467776f8fa93 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,