diff --git a/config/AEPSA/NYCBike-inflow.yaml b/config/AEPSA/NYCBike-inflow.yaml
index fd5ed41..4cd9d3b 100644
--- a/config/AEPSA/NYCBike-inflow.yaml
+++ b/config/AEPSA/NYCBike-inflow.yaml
@@ -1,5 +1,5 @@
 basic:
-  dataset: "NYCBike-inflow"
+  dataset: "NYCBike-InFlow"
   mode : "train"
   device : "cuda:0"
   model: "AEPSA"
diff --git a/config/AEPSA/NYCBike-outflow.yaml b/config/AEPSA/NYCBike-outflow.yaml
index ba7b014..cf9b696 100644
--- a/config/AEPSA/NYCBike-outflow.yaml
+++ b/config/AEPSA/NYCBike-outflow.yaml
@@ -1,5 +1,5 @@
 basic:
-  dataset: "NYCBike-outflow"
+  dataset: "NYCBike-OutFlow"
   mode : "train"
   device : "cuda:0"
   model: "AEPSA"