diff --git a/datasets/biomassters.py b/datasets/biomassters.py
index cd7aae80..b2244c17 100644
--- a/datasets/biomassters.py
+++ b/datasets/biomassters.py
@@ -10,10 +10,10 @@
 
 def read_imgs(multi_temporal, temp , fname, data_dir, img_size):
     imgs_s1, imgs_s2, mask = [], [], []
-    if multi_temporal:
-        month_list = list(range(12))
+    if multi_temporal==1:
+        month_list = [temp]        
     else:
-        month_list = [temp]
+        month_list = list(range(12))
     
     for month in month_list: