diff --git a/src/main/java/com/kamco/cd/training/train/dto/TrainRunRequest.java b/src/main/java/com/kamco/cd/training/train/dto/TrainRunRequest.java
index e294ce7..5761131 100644
--- a/src/main/java/com/kamco/cd/training/train/dto/TrainRunRequest.java
+++ b/src/main/java/com/kamco/cd/training/train/dto/TrainRunRequest.java
@@ -15,7 +15,7 @@ public class TrainRunRequest {
   // ========================
   // 기본
   // ========================
-  private UUID datasetFolder;
+  private String datasetFolder;
   private UUID outputFolder;
   private String inputSize;
   private String cropSize;
@@ -84,10 +84,6 @@ public class TrainRunRequest {
 
   private UUID uuid;
 
-  public String getDatasetFolder() {
-    return String.valueOf(this.datasetFolder);
-  }
-
   public String getOutputFolder() {
     return String.valueOf(this.outputFolder);
   }