diff --git a/PWGHF/D2H/Tasks/taskDplus.cxx b/PWGHF/D2H/Tasks/taskDplus.cxx
index b09e705d3ab..2399b6c99a6 100644
--- a/PWGHF/D2H/Tasks/taskDplus.cxx
+++ b/PWGHF/D2H/Tasks/taskDplus.cxx
@@ -230,7 +230,7 @@ struct HfTaskDplus {
       if (storeIR) {
         axes.push_back(thnAxisIR);
       }
-      if (doprocessDataWithMlWithUpc) {
+      if (doprocessDataWithMlWithUpc || doprocessDataWithUpc) {
         axes.push_back(thnAxisGapType);
         axes.push_back(thnAxisFT0A);
         axes.push_back(thnAxisFT0C);