diff --git a/trained_model_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml b/trained_model_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml
index cae1ac81..0bab0d90 100644
--- a/trained_model_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml
+++ b/trained_model_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml
@@ -244,5 +244,5 @@ predict:
     interpolation_k: 10
     classification_dict: ${dataset_description.classification_dict}
     probas_to_save: [building,ground]
-    predicted_classification_channel: reliability
+    predicted_classification_channel: confidence
     entropy_channel: entropy