diff --git a/baselines/STID/GBA.py b/baselines/STID/GBA.py
index 964e6925..6ef41121 100644
--- a/baselines/STID/GBA.py
+++ b/baselines/STID/GBA.py
@@ -14,7 +14,7 @@
 
 ############################## Hot Parameters ##############################
 # Dataset & Metrics configuration
-DATA_NAME = 'CA'  # Dataset name
+DATA_NAME = 'GBA'  # Dataset name
 regular_settings = get_regular_settings(DATA_NAME)
 INPUT_LEN = regular_settings['INPUT_LEN']  # Length of input sequence
 OUTPUT_LEN = regular_settings['OUTPUT_LEN']  # Length of output sequence
diff --git a/baselines/STID/GLA.py b/baselines/STID/GLA.py
index 7e997b05..6d8c1c09 100644
--- a/baselines/STID/GLA.py
+++ b/baselines/STID/GLA.py
@@ -14,7 +14,7 @@
 
 ############################## Hot Parameters ##############################
 # Dataset & Metrics configuration
-DATA_NAME = 'CA'  # Dataset name
+DATA_NAME = 'GLA'  # Dataset name
 regular_settings = get_regular_settings(DATA_NAME)
 INPUT_LEN = regular_settings['INPUT_LEN']  # Length of input sequence
 OUTPUT_LEN = regular_settings['OUTPUT_LEN']  # Length of output sequence