From 9ed2ec27697e428242f2e3b33b5c5335562c6d84 Mon Sep 17 00:00:00 2001
From: Mingze Li <mingze-li@foxmail.com>
Date: Sat, 25 Apr 2026 09:41:35 +0200
Subject: [PATCH] Fix THnSparseCosThStarEP axis configuration

---
 PWGHF/D2H/Tasks/taskCharmPolarisation.cxx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/PWGHF/D2H/Tasks/taskCharmPolarisation.cxx b/PWGHF/D2H/Tasks/taskCharmPolarisation.cxx
index 110669b5810..74a57633ce1 100644
--- a/PWGHF/D2H/Tasks/taskCharmPolarisation.cxx
+++ b/PWGHF/D2H/Tasks/taskCharmPolarisation.cxx
@@ -794,8 +794,8 @@ struct HfTaskCharmPolarisation {
       LOGP(fatal, "THnSparse with cosThStar w.r.t. event plane axis is not supported for pp analysis, please check the configuration!");
     } else if (activateTHnSparseCosThStarEP) {
       std::vector<AxisSpec> hEPaxes = {thnAxisInvMass, thnAxisPt, thnAxisNumPvContributors, thnAxisY};
+      hEPaxes.insert(hEPaxes.end(), {thnAxisInvMassD0, thnAxisCosThetaStarEP});
       if (doprocessDstarInPbPb || doprocessDstarWithMlInPbPb) {
-        hEPaxes.insert(hEPaxes.end(), {thnAxisInvMassD0, thnAxisCosThetaStarEP});
         if (doprocessDstarWithMlInPbPb) {
           hEPaxes.insert(hEPaxes.end(), {thnAxisMlBkg, thnAxisMlNonPrompt});
         }