From 1265ce66204f5a7c7c33cae9666b0117b5d4bb42 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?BiaoZhang=20=28=E5=BC=A0=E5=BD=AA=29?=
 <52267892+zhangbiao-phy@users.noreply.github.com>
Date: Tue, 25 Mar 2025 15:57:33 +0100
Subject: [PATCH] Update treeCreatorTccToD0D0Pi.cxx

---
 PWGHF/TableProducer/treeCreatorTccToD0D0Pi.cxx | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/PWGHF/TableProducer/treeCreatorTccToD0D0Pi.cxx b/PWGHF/TableProducer/treeCreatorTccToD0D0Pi.cxx
index e50597e7c9f..3d4f12dba49 100644
--- a/PWGHF/TableProducer/treeCreatorTccToD0D0Pi.cxx
+++ b/PWGHF/TableProducer/treeCreatorTccToD0D0Pi.cxx
@@ -291,8 +291,8 @@ struct HfTreeCreatorTccToD0D0Pi {
             candFlagD1 = (candidateD1.isSelD0bar()) ? 3 : 1;
             std::copy(candidateD1.mlProbD0().begin(), candidateD1.mlProbD0().end(), std::back_inserter(mlScoresD1));
             massD01 = hfHelper.invMassD0ToPiK(candidateD1);
-          } else if (candidateD1.isSelD0bar()) {
-            candFlagD1 = (candidateD1.isSelD0()) ? 4 : 2;
+          }
+          if (candidateD1.isSelD0bar()) {
             std::copy(candidateD1.mlProbD0bar().begin(), candidateD1.mlProbD0bar().end(), std::back_inserter(mlScoresD1));
             massD01 = hfHelper.invMassD0barToKPi(candidateD1);
           }
@@ -301,9 +301,8 @@ struct HfTreeCreatorTccToD0D0Pi {
             candFlagD2 = (candidateD2.isSelD0bar()) ? 3 : 1;
             std::copy(candidateD2.mlProbD0().begin(), candidateD2.mlProbD0().end(), std::back_inserter(mlScoresD2));
             massD02 = hfHelper.invMassD0ToPiK(candidateD2);
-
-          } else if (candidateD2.isSelD0bar()) {
-            candFlagD2 = (candidateD2.isSelD0()) ? 4 : 2;
+          }
+          if (candidateD2.isSelD0bar()) {
             std::copy(candidateD2.mlProbD0bar().begin(), candidateD2.mlProbD0bar().end(), std::back_inserter(mlScoresD2));
             massD02 = hfHelper.invMassD0barToKPi(candidateD2);
           }