diff --git a/configs/synthesizers/UGATIT.yml b/configs/synthesizers/UGATIT.yml
index 07160fe..05d5311 100644
--- a/configs/synthesizers/UGATIT.yml
+++ b/configs/synthesizers/UGATIT.yml
@@ -103,9 +103,10 @@ data:
       pin_memory: False
       drop_last: False
     dataset:
-      _type: SingleFolderDataset
-      root: "path/to/images/"
-      with_path: True
+      _type: GenerationUnpairedDataset
+      root_a: "/data/i2i/selfie2anime/testA"
+      root_b: "/data/i2i/selfie2anime/testB"
+      random_pair: False
       pipeline:
         - Load
         - Resize: