diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 457b524dc0ae7cc60a68d148ed304af4ee426ba9..c3066ad4717d46917343517ab25890709431dfaf 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2522,8 +2522,8 @@ class NetworkInference(BaseConfig):
                               (batch + 1) * self.dataloader.batch_size)]
 
                 # name of the current scene
-                batch = np.unique([sid for sid in self.trg_ds.dataset.scenes[
-                    current_tiles]['id']]).item()
+                batch = np.unique([self.trg_ds.dataset.scenes[sid]['id'] for
+                                   sid in current_tiles]).item()
 
                 # modify the progress string
                 progress = progress.replace('Sample', 'Scene')