diff --git a/src/main/java/com/kamco/cd/training/train/service/TrainJobWorker.java b/src/main/java/com/kamco/cd/training/train/service/TrainJobWorker.java
index 44dbe1d..5e17a1a 100644
--- a/src/main/java/com/kamco/cd/training/train/service/TrainJobWorker.java
+++ b/src/main/java/com/kamco/cd/training/train/service/TrainJobWorker.java
@@ -100,10 +100,10 @@ public class TrainJobWorker {
 
         if (isEval) {
           modelTrainMngCoreService.markStep2Success(modelId);
-          modelTrainMetricsJobService.findTrainValidMetricCsvFiles();
+          modelTestMetricsJobService.findTestValidMetricCsvFiles();
         } else {
           modelTrainMngCoreService.markStep1Success(modelId);
-          modelTestMetricsJobService.findTestValidMetricCsvFiles();
+          modelTrainMetricsJobService.findTrainValidMetricCsvFiles();
         }
 
       } else {