diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index f642e05cba63971b6c8e74ceaf6892e8be8bb9f2..d8dfc714e9ed539f7a5b404573e363a603a5943b 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -754,7 +754,7 @@ class StandardEoDataset(ImageDataset):
 
                 # list the spectral bands of the scene
                 bands = [os.path.join(dirpath, f) for f in files
-                         if re.search('B\\dA|B\\d{1,2}', f)]
+                         if re.search('B\\dA|B\\d{1,2}(.*).tif$', f)]
 
                 # get the ground truth mask
                 try: