From 20abd2cb667c705e005243216f929779cf8b3adc 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: Fri, 2 Jan 2026 12:41:46 +0100
Subject: [PATCH] Update taskFlowCharmHadrons.cxx

---
 PWGHF/D2H/Tasks/taskFlowCharmHadrons.cxx | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/PWGHF/D2H/Tasks/taskFlowCharmHadrons.cxx b/PWGHF/D2H/Tasks/taskFlowCharmHadrons.cxx
index 7c2298c3611..40c253d051d 100644
--- a/PWGHF/D2H/Tasks/taskFlowCharmHadrons.cxx
+++ b/PWGHF/D2H/Tasks/taskFlowCharmHadrons.cxx
@@ -530,11 +530,10 @@ struct HfTaskFlowCharmHadrons {
       float signCand = 0.;
       std::vector<float> outputMl = {-999., -999.};
       if constexpr (std::is_same_v<T1, CandDsData> || std::is_same_v<T1, CandDsDataWMl>) {
-        auto trackprong0 = candidate.template prong0_as<Trk>();
-        signCand = trackprong0.sign();
         switch (Channel) {
           case DecayChannel::DsToKKPi:
             massCand = HfHelper::invMassDsToKKPi(candidate);
+            signCand = 1;
             if constexpr (std::is_same_v<T1, CandDsDataWMl>) {
               for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {
                 outputMl[iclass] = candidate.mlProbDsToKKPi()[classMl->at(iclass)];
@@ -543,6 +542,7 @@ struct HfTaskFlowCharmHadrons {
             break;
           case DecayChannel::DsToPiKK:
             massCand = HfHelper::invMassDsToPiKK(candidate);
+            signCand = -1;
             if constexpr (std::is_same_v<T1, CandDsDataWMl>) {
               for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {
                 outputMl[iclass] = candidate.mlProbDsToPiKK()[classMl->at(iclass)];
@@ -585,11 +585,10 @@ struct HfTaskFlowCharmHadrons {
             break;
         }
       } else if constexpr (std::is_same_v<T1, CandLcData> || std::is_same_v<T1, CandLcDataWMl>) {
-        auto trackprong0 = candidate.template prong0_as<Trk>();
-        signCand = trackprong0.sign();
         switch (Channel) {
           case DecayChannel::LcToPKPi:
             massCand = HfHelper::invMassLcToPKPi(candidate);
+            signCand = 1;
             if constexpr (std::is_same_v<T1, CandLcDataWMl>) {
               for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {
                 outputMl[iclass] = candidate.mlProbLcToPKPi()[classMl->at(iclass)];
@@ -598,6 +597,7 @@ struct HfTaskFlowCharmHadrons {
             break;
           case DecayChannel::LcToPiKP:
             massCand = HfHelper::invMassLcToPiKP(candidate);
+            signCand = -1;
             if constexpr (std::is_same_v<T1, CandLcDataWMl>) {
               for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {
                 outputMl[iclass] = candidate.mlProbLcToPiKP()[classMl->at(iclass)];
@@ -608,11 +608,10 @@ struct HfTaskFlowCharmHadrons {
             break;
         }
       } else if constexpr (std::is_same_v<T1, CandXicData> || std::is_same_v<T1, CandXicDataWMl>) {
-        auto trackprong0 = candidate.template prong0_as<Trk>();
-        signCand = trackprong0.sign();
         switch (Channel) {
           case DecayChannel::XicToPKPi:
             massCand = HfHelper::invMassXicToPKPi(candidate);
+            signCand = 1;
             if constexpr (std::is_same_v<T1, CandXicDataWMl>) {
               for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {
                 outputMl[iclass] = candidate.mlProbXicToPKPi()[classMl->at(iclass)];
@@ -621,6 +620,7 @@ struct HfTaskFlowCharmHadrons {
             break;
           case DecayChannel::XicToPiKP:
             massCand = HfHelper::invMassXicToPiKP(candidate);
+            signCand = -1;
             if constexpr (std::is_same_v<T1, CandXicDataWMl>) {
               for (unsigned int iclass = 0; iclass < classMl->size(); iclass++) {
                 outputMl[iclass] = candidate.mlProbXicToPiKP()[classMl->at(iclass)];