diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py
index e2161efa875052920e0eecea60c0ca2f9319f546..b2d9d0733a22b402e18d37a052bf20007c129490 100644
--- a/modules/deeplearning/cloud_fraction_fcn_abi.py
+++ b/modules/deeplearning/cloud_fraction_fcn_abi.py
@@ -184,11 +184,11 @@ def get_label_data_5cat(grd_k):
     d = grd_k[:, 1::4, 1::4]
     s = a + b + c + d
 
-    cat_0 = (s <= 1)
-    cat_1 = (s > 1) & (s <= 4)
-    cat_2 = (s > 4) & (s <= 11)
-    cat_3 = (s > 11) & (s <= 14)
-    cat_4 = (s > 14)
+    cat_0 = np.logical_and(s >= 0, s < 2)
+    cat_1 = np.logical_and(s >= 2, s < 6)
+    cat_2 = np.logical_and(s >= 6, s < 11)
+    cat_3 = np.logical_and(s >= 11, s < 15)
+    cat_4 = np.logical_and(s >= 15, s <= 16)
 
     s[cat_0] = 0
     s[cat_1] = 1