diff --git a/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py b/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py
index 5e29e2e21..f29d29f16 100644
--- a/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py
+++ b/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py
@@ -46,6 +46,7 @@ def train(self, tree_num, node_num=None, training_info=None):
 
     def _check_eval_finish(self, tree_num):
         if self.eval_finish_flag:
+            self.eval_finish_flag = False
             if tree_num + 1 < self._cfg.model.num_of_trees:
                 batch_index, feature_order_info = \
                     self.trainer.fetch_train_data()