diff --git a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
index 60ee90d..e33e83d 100644
--- a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
@@ -62,7 +62,7 @@ public class DockerTrainService {
   @Value("${hyper.parameter.gpu-ids}")
   private String hyperGpuIds;
 
-  @Value("${hyper.parameter.batch_size}")
+  @Value("${hyper.parameter.batch-size}")
   private Integer batchSize;
 
   private final ModelTrainJobCoreService modelTrainJobCoreService;