diff --git a/climax/main/downscale_infer.py b/climax/main/downscale_infer.py
index eedd85aec282d5e85a526042abf9f3eaaf596a27..0a7945fb1204c6770660e6f6393c383d88e63616 100644
--- a/climax/main/downscale_infer.py
+++ b/climax/main/downscale_infer.py
@@ -47,8 +47,8 @@ if __name__ == '__main__':
     # path to model state
     if SENSITIVITY:
         # models trained for hyperparameter optimization
-        state_file = MODEL_PATH.joinpath(PREDICTAND, 'sensitivity', state_file)
-        target = TARGET_PATH.joinpath(PREDICTAND, 'sensitivity')
+        state_file = MODEL_PATH.joinpath('sensitivity', PREDICTAND, state_file)
+        target = TARGET_PATH.joinpath('sensitivity', PREDICTAND)
     else:
         state_file = MODEL_PATH.joinpath(PREDICTAND, state_file)
         target = TARGET_PATH.joinpath(PREDICTAND)
diff --git a/climax/main/downscale_train.py b/climax/main/downscale_train.py
index 9dc09aba53dff5758cfa34af12973858e491353f..87e994bd54c3628e0293307ca0d8dc6088c3e095 100644
--- a/climax/main/downscale_train.py
+++ b/climax/main/downscale_train.py
@@ -49,7 +49,7 @@ if __name__ == '__main__':
     # path to model state
     if SENSITIVITY:
         # models trained for hyperparameter optimization
-        state_file = MODEL_PATH.joinpath(PREDICTAND, 'sensitivity', state_file)
+        state_file = MODEL_PATH.joinpath('sensitivity', PREDICTAND, state_file)
     else:
         state_file = MODEL_PATH.joinpath(PREDICTAND, state_file)