diff --git a/benchmarks/imaging_shower_ML/config.yml b/benchmarks/imaging_shower_ML/config.yml
index bccac7e7491b8c022c847603c071ac366d598fb9..2e16d1f13f5519a3ae39cb05247e9858c45878bd 100644
--- a/benchmarks/imaging_shower_ML/config.yml
+++ b/benchmarks/imaging_shower_ML/config.yml
@@ -27,7 +27,7 @@ ml_shower:training:
   extends: .rec_benchmark
   when: manual
   stage: process
-  needs: ["ml_shower:tagging"]
+  needs: ["ml_shower:tagging_epi", "ml_shower:tagging_e", "ml_shower:tagging_pi"]
   script:
     - ls -lrth
     # TODO