diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index a32fa26a406a573ea31ed5a9725b239e5549b665..12cdafbc822843973eeae49db4d93fce1e6572ed 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2314,7 +2314,7 @@ class NetworkInference(BaseConfig):
             identifiers used for evaluation.
 
         """
-        return np.array([list(self._original_target_labels_labels.keys()),
+        return np.array([list(self._original_target_labels.keys()),
                          list(self.target_labels.keys())]).T
 
     @property