diff --git a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/configs/subvpsde/xarray_cncsnpp_continuous.py b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/configs/subvpsde/xarray_cncsnpp_continuous.py
index b41907d72..c3a2b22b7 100644
--- a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/configs/subvpsde/xarray_cncsnpp_continuous.py
+++ b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/configs/subvpsde/xarray_cncsnpp_continuous.py
@@ -64,5 +64,7 @@ def get_config():
 
   # data
   data = config.data
+  data.input_transform_key = "stan"
+  data.target_transform_key = "sqrturrecen"
 
   return config