diff --git a/src/MEMClassifier.cc b/src/MEMClassifier.cc index d06629c5797db161f0d63afcc45579b9213036b2..0b2bf689b8fadc81aed63a0c424ffe83c260e7bf 100644 --- a/src/MEMClassifier.cc +++ b/src/MEMClassifier.cc @@ -467,8 +467,10 @@ TH3D* MEMClassifier::GetBTagPDF(const char* prefix, const char* flavour) { double MEMClassifier::GetJetBProbability(const char* prefix, const char* flavour, double pt, double eta, double bdisc) { TH3D* h = GetBTagPDF(prefix, flavour); - int i = h->FindBin(pt, std::abs(eta), bdisc); - return h->GetBinContent(i); + const int i = h->FindBin(pt, std::abs(eta), bdisc); + const double c = h->GetBinContent(i); + delete h; + return c; } MEM::JetProbability MEMClassifier::GetJetBProbabilities(