diff --git a/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py b/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py
index 0a40546a3..32d078b28 100644
--- a/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py
+++ b/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py
@@ -178,6 +178,7 @@ def callback_funcs_for_local_best_gain(self, message: Message):
                     self.comm_manager.send(send_message)
             else:
                 self.trainer._set_weight_and_status(tree_num, node_num)
+                self.train(tree_num=tree_num, node_num=node_num + 1)
 
     # Bind method to instance
     client.train = types.MethodType(train, client)