improve the MC truth matching