diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 792ebe20cbb5bf39d92349d05cfd72c70d32bb69..e663a336f417f73fa9649dae8141503159899859 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2784,7 +2784,7 @@ class NetworkInference(BaseConfig):
                 # check if for some scenes micro average is calculated rather
                 # than accuracy due to missing class labels
                 labels = list(report.index.drop(['macro avg', 'weighted avg',
-                                                 'micro_avg', 'accuracy']))
+                                                 'micro avg', 'accuracy']))
             except KeyError:
                 # micro average is not calculated for any scene: each scene
                 # contains all the classes