diff --git a/PWGEM/Dilepton/Utils/MlResponseDielectronSingleTrack.h b/PWGEM/Dilepton/Utils/MlResponseDielectronSingleTrack.h
index 2ed0b02fe3c..5ec9756b5a9 100644
--- a/PWGEM/Dilepton/Utils/MlResponseDielectronSingleTrack.h
+++ b/PWGEM/Dilepton/Utils/MlResponseDielectronSingleTrack.h
@@ -144,6 +144,7 @@ enum class InputFeaturesDielectronSingleTrack : uint8_t {
   tgl,
   isAssociatedToMPC,
   tpcNClsFound,
+  tpcNClsPID,
   tpcNClsCrossedRows,
   tpcCrossedRowsOverFindableCls,
   tpcFoundOverFindableCls,
@@ -266,6 +267,7 @@ class MlResponseDielectronSingleTrack : public MlResponse<TypeOutputScore>
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tgl);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(isAssociatedToMPC);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcNClsFound);
+      CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcNClsPID);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcNClsCrossedRows);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcCrossedRowsOverFindableCls);
       CHECK_AND_FILL_DIELECTRON_SINGLE_TRACK(tpcFoundOverFindableCls);
@@ -414,6 +416,7 @@ class MlResponseDielectronSingleTrack : public MlResponse<TypeOutputScore>
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tgl),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(isAssociatedToMPC),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcNClsFound),
+      FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcNClsPID),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcNClsCrossedRows),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcCrossedRowsOverFindableCls),
       FILL_MAP_DIELECTRON_SINGLE_TRACK(tpcFoundOverFindableCls),
diff --git a/PWGEM/Dilepton/Utils/MlResponseO2Track.h b/PWGEM/Dilepton/Utils/MlResponseO2Track.h
index 5e1f2fc4226..d898d9da5f6 100644
--- a/PWGEM/Dilepton/Utils/MlResponseO2Track.h
+++ b/PWGEM/Dilepton/Utils/MlResponseO2Track.h
@@ -160,6 +160,7 @@ enum class InputFeaturesO2Track : uint8_t {
   tpctofNSigmaKa,
   tpctofNSigmaPr,
   tpcNClsFound,
+  tpcNClsPID,
   tpcNClsCrossedRows,
   tpcChi2NCl,
   hasITS,
@@ -215,6 +216,7 @@ class MlResponseO2Track : public MlResponse<TypeOutputScore>
       CHECK_AND_FILL_O2_TRACK_TPCTOF(tpctofNSigmaKa, tpcNSigmaKa, tofNSigmaKa, hasTOF);
       CHECK_AND_FILL_O2_TRACK_TPCTOF(tpctofNSigmaPr, tpcNSigmaPr, tofNSigmaPr, hasTOF);
       CHECK_AND_FILL_O2_TRACK(tpcNClsFound);
+      CHECK_AND_FILL_O2_TRACK(tpcNClsPID);
       CHECK_AND_FILL_O2_TRACK(tpcNClsCrossedRows);
       CHECK_AND_FILL_O2_TRACK(tpcChi2NCl);
       CHECK_AND_FILL_O2_TRACK(hasITS);
@@ -296,6 +298,7 @@ class MlResponseO2Track : public MlResponse<TypeOutputScore>
       FILL_MAP_O2_TRACK(tpctofNSigmaKa),
       FILL_MAP_O2_TRACK(tpctofNSigmaPr),
       FILL_MAP_O2_TRACK(tpcNClsFound),
+      FILL_MAP_O2_TRACK(tpcNClsPID),
       FILL_MAP_O2_TRACK(tpcNClsCrossedRows),
       FILL_MAP_O2_TRACK(tpcChi2NCl),
       FILL_MAP_O2_TRACK(hasITS),