diff --git a/src/main/java/org/grobid/trainer/UnitTrainer.java b/src/main/java/org/grobid/trainer/UnitTrainer.java
index e52379f3..0dce7957 100644
--- a/src/main/java/org/grobid/trainer/UnitTrainer.java
+++ b/src/main/java/org/grobid/trainer/UnitTrainer.java
@@ -161,6 +161,7 @@ public boolean accept(File dir, String name) {
 
                     addFeatures(labels, writer, unitTokenPositions, labeledUnit.isUnitLeft());
                     writer.write("\n");
+                    writer = dispatchExample(trainingOutputWriter, evaluationOutputWriter, splitRatio);
                     pos++;
                 }
             }
diff --git a/src/main/java/org/grobid/trainer/ValueTrainer.java b/src/main/java/org/grobid/trainer/ValueTrainer.java
index 063e8bf1..1b51ebab 100644
--- a/src/main/java/org/grobid/trainer/ValueTrainer.java
+++ b/src/main/java/org/grobid/trainer/ValueTrainer.java
@@ -150,6 +150,7 @@ public boolean accept(File dir, String name) {
 
                     addFeatures(labels, writer);
                     writer.write("\n");
+                    writer = dispatchExample(trainingOutputWriter, evaluationOutputWriter, splitRatio);
                     pos++;
                 }
             }