From eb81b375f254dadc2cb807587327825cc64c1723 Mon Sep 17 00:00:00 2001
From: Ravindra Singh <56298081+1994ra@users.noreply.github.com>
Date: Tue, 10 Dec 2024 20:21:17 +0100
Subject: [PATCH] [PWGHF] PID inclusion for associated particles in Lc
 correlation

---
 PWGHF/HFC/Tasks/taskCorrelationLcHadrons.cxx | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/PWGHF/HFC/Tasks/taskCorrelationLcHadrons.cxx b/PWGHF/HFC/Tasks/taskCorrelationLcHadrons.cxx
index cf6c8f3772f..ede013d1f22 100644
--- a/PWGHF/HFC/Tasks/taskCorrelationLcHadrons.cxx
+++ b/PWGHF/HFC/Tasks/taskCorrelationLcHadrons.cxx
@@ -641,13 +641,17 @@ struct HfTaskCorrelationLcHadrons {
 
       // fill correlation plots for signal/bagkground correlations
       if (pairEntry.signalStatus()) {
-        registry.fill(HIST("hCorrel2DVsPtSignalMcRec"), deltaPhi, deltaEta, ptLc, ptHadron, poolBin, efficiencyWeight);
-        if (fillSign)
+        if (fillSign) {
           registry.fill(HIST("hCorrel2DVsPtSignSignalMcRec"), deltaPhi, deltaEta, ptLc, ptHadron, signPair, poolBin, efficiencyWeight);
+        } else {
+          registry.fill(HIST("hCorrel2DVsPtSignalMcRec"), deltaPhi, deltaEta, ptLc, ptHadron, poolBin, efficiencyWeight);
+        }
       } else {
-        registry.fill(HIST("hCorrel2DVsPtBkgMcRec"), deltaPhi, deltaEta, ptLc, ptHadron, poolBin, efficiencyWeight);
-        if (fillSign)
+        if (fillSign) {
           registry.fill(HIST("hCorrel2DVsPtSignBkgMcRec"), deltaPhi, deltaEta, ptLc, ptHadron, signPair, poolBin, efficiencyWeight);
+        } else {
+          registry.fill(HIST("hCorrel2DVsPtBkgMcRec"), deltaPhi, deltaEta, ptLc, ptHadron, poolBin, efficiencyWeight);
+        }
       }
       // reject entries outside Pt ranges of interest