adapt lepton CR to new trees