diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py
index 46d16f24b4dedf7eec72345e04a7cb1c7255ed35..02bdc61864f16dc5d415912f59ad923c3b20c2ce 100644
--- a/modules/deeplearning/cloud_fraction_fcn_abi.py
+++ b/modules/deeplearning/cloud_fraction_fcn_abi.py
@@ -178,11 +178,10 @@ def get_label_data_5cat(grd_k):
     grd_k = np.where(np.isnan(grd_k), 0, grd_k)
     grd_k = np.where(grd_k < 0.5, 0, 1)
 
-    a = grd_k[:, 0::4, 0::4]
-    b = grd_k[:, 1::4, 0::4]
-    c = grd_k[:, 0::4, 1::4]
-    d = grd_k[:, 1::4, 1::4]
-    s = a + b + c + d
+    s = grd_k[:, 0::4, 0::4] + grd_k[:, 1::4, 0::4] + grd_k[:, 2::4, 0::4] + grd_k[:, 3::4, 0::4] + \
+        grd_k[:, 0::4, 1::4] + grd_k[:, 1::4, 1::4] + grd_k[:, 2::4, 1::4] + grd_k[:, 3::4, 1::4] + \
+        grd_k[:, 0::4, 2::4] + grd_k[:, 1::4, 2::4] + grd_k[:, 2::4, 2::4] + grd_k[:, 3::4, 2::4] + \
+        grd_k[:, 0::4, 3::4] + grd_k[:, 1::4, 3::4] + grd_k[:, 2::4, 3::4] + grd_k[:, 3::4, 3::4]
 
     cat_0 = np.logical_and(s >= 0, s < 2)
     cat_1 = np.logical_and(s >= 2, s < 6)