diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index d09a6c31ee3c18739bd7778a0122422ced8e5d3e..9105af4d2e4bbc38c5ef74b5d39f3f5e646be418 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2546,7 +2546,7 @@ class NetworkInference(BaseConfig):
                                 self.bands,
                                 self.source_labels,
                                 y=labels,
-                                y_pred={self.model.__class__.__name__: prdctn},
+                                y_pred={model.__class__.__name__: prdctn},
                                 accuracy=True,
                                 state=batch_name,
                                 plot_path=self.scenes_path,