diff --git a/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java b/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
index 08c01fe..1edfe52 100644
--- a/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
+++ b/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
@@ -66,7 +66,7 @@ public class TmpDatasetService {
       Path srcRoot = BASE.resolve(id);
       log.info("---- dataset id={} srcRoot={} exists? {}", id, srcRoot, Files.isDirectory(srcRoot));
 
-      for (String type : List.of("train", "val")) {
+      for (String type : List.of("train", "val", "test")) {
         for (String part : List.of("input1", "input2", "label")) {
 
           Path srcDir = srcRoot.resolve(type).resolve(part);