Fix vertex distribution before PU reweighting