diff --git a/config/REPST/AirQuality.yaml b/config/REPST/AirQuality.yaml
index ee690f1..8eeba4f 100755
--- a/config/REPST/AirQuality.yaml
+++ b/config/REPST/AirQuality.yaml
@@ -27,7 +27,7 @@ model:
   input_dim: 6
   n_heads: 1
   num_nodes: 12
-  output_dim: 3
+  output_dim: 6
   patch_len: 6
   pred_len: 24
   seq_len: 24