diff --git a/PWGHF/Core/HfMlResponseOmegacToOmegaPi.h b/PWGHF/Core/HfMlResponseOmegacToOmegaPi.h
index d3afc762d83..13e4731f491 100644
--- a/PWGHF/Core/HfMlResponseOmegacToOmegaPi.h
+++ b/PWGHF/Core/HfMlResponseOmegacToOmegaPi.h
@@ -74,6 +74,7 @@ enum class InputFeaturesOmegacToOmegaPi : uint8_t {
   dcaCascDau,
   cosPaCascToOmegac,
   decayLenXYCasc,
+  decayLenXYOmegac,
   ldlOmegac,
   chi2NdfTopoCascToOmegac,
   chi2NdfTopoCascToPv,
@@ -120,6 +121,7 @@ class HfMlResponseOmegacToOmegaPi : public HfMlResponse<TypeOutputScore>
         CHECK_AND_FILL_VEC_OMEGAC0(dcaCascDau);
         CHECK_AND_FILL_VEC_OMEGAC0(cosPaCascToOmegac);
         CHECK_AND_FILL_VEC_OMEGAC0(decayLenXYCasc);
+        CHECK_AND_FILL_VEC_OMEGAC0(decayLenXYOmegac);
         CHECK_AND_FILL_VEC_OMEGAC0_FULL(candidate, ldlOmegac, omegacldl);
         CHECK_AND_FILL_VEC_OMEGAC0_FULL(candidate, chi2NdfTopoCascToOmegac, chi2TopoCascToOmegac);
         CHECK_AND_FILL_VEC_OMEGAC0_FULL(candidate, chi2NdfTopoCascToPv, chi2TopoCascToPv);
@@ -156,6 +158,7 @@ class HfMlResponseOmegacToOmegaPi : public HfMlResponse<TypeOutputScore>
       FILL_MAP_OMEGAC0(dcaCascDau),
       FILL_MAP_OMEGAC0(cosPaCascToOmegac),
       FILL_MAP_OMEGAC0(decayLenXYCasc),
+      FILL_MAP_OMEGAC0(decayLenXYOmegac),
       FILL_MAP_OMEGAC0(ldlOmegac),
       FILL_MAP_OMEGAC0(chi2NdfTopoCascToOmegac),
       FILL_MAP_OMEGAC0(chi2NdfTopoCascToPv),