diff --git a/PWGEM/Dilepton/Utils/MlResponsePID.h b/PWGEM/Dilepton/Utils/MlResponsePID.h
index 63e85a6b063..13e8048f798 100644
--- a/PWGEM/Dilepton/Utils/MlResponsePID.h
+++ b/PWGEM/Dilepton/Utils/MlResponsePID.h
@@ -63,7 +63,7 @@ struct candidate {
   float tofNSigmaEl{0};
   float meanClusterSizeITSobCosTgl{0};
 };
-} // namespace pwgem::dilepton
+} // namespace pwgem::dilepton::mlpid
 
 template <typename TypeOutputScore = float>
 class MlResponsePID : public MlResponse<TypeOutputScore>
diff --git a/PWGEM/Dilepton/Utils/MlResponseSCT.h b/PWGEM/Dilepton/Utils/MlResponseSCT.h
index 71037655e85..d0c1353e443 100644
--- a/PWGEM/Dilepton/Utils/MlResponseSCT.h
+++ b/PWGEM/Dilepton/Utils/MlResponseSCT.h
@@ -124,7 +124,7 @@ struct candidate {
   float decayLengthZinSigma{0};
   float decayLength3DinSigma{0};
 };
-} // namespace pwgem::dilepton
+} // namespace pwgem::dilepton::sct
 
 template <typename TypeOutputScore = float>
 class MlResponseSCT : public MlResponse<TypeOutputScore>