diff --git a/modules/util/viirs_l1b_l2.py b/modules/util/viirs_l1b_l2.py
index c23a8c3de67647a4c05c74a29f47059785d7efd7..55810b0e3d7e4f3a41987aacf683a55c2b6eb9d1 100644
--- a/modules/util/viirs_l1b_l2.py
+++ b/modules/util/viirs_l1b_l2.py
@@ -237,9 +237,12 @@ def run(data_h5f, param_s, train_tiles, valid_tiles, num_keep_x_tiles=8, tile_wi
                 continue
 
             nda = data[:, j_a:j_b, i_a:i_b]
-            if not keep_tile(param_s, nda):
-                continue
-            train_tiles.append(nda)
+            # if not keep_tile(param_s, nda):
+            #     continue
+
+            nda = process_cld_prob(param_s, nda)
+            if nda is not None:
+                train_tiles.append(nda)
 
     j_start = num_y_train * tile_width + 2*tile_width
     for j in range(num_y_valid):