diff --git a/PWGEM/Dilepton/Tasks/mcParticlePredictionsOTF.cxx b/PWGEM/Dilepton/Tasks/mcParticlePredictionsOTF.cxx
index 4ebd94f90a1..7d7679b42d9 100644
--- a/PWGEM/Dilepton/Tasks/mcParticlePredictionsOTF.cxx
+++ b/PWGEM/Dilepton/Tasks/mcParticlePredictionsOTF.cxx
@@ -32,8 +32,7 @@ struct otfParticlePrediction {
   Configurable<float> maxEtaParticle{"maxEtaParticle", 5.f, "Max eta of particles considered"};
 
   // init function
-  void
-    init()
+  void init(InitContext&)
   {
 
     const AxisSpec axisEta{binsEta, "#eta"};
@@ -56,12 +55,12 @@ struct otfParticlePrediction {
 
     for (const auto& particle : mcParticles) {
       auto pdg = std::abs(particle.pdgCode());
-      if (particle.eta() < maxEtaParticle) {
-        continue;
-      }
-      if (!particle.isPhysicalPrimary()) {
+      if (std::abs(particle.eta()) > maxEtaParticle) {
         continue;
       }
+      // if (!(particle.isPhysicalPrimary())) {
+      //   continue;
+      // }
       if (pdg < 100) {
         continue;
       }