diff --git a/modules/deeplearning/cnn_cld_frac.py b/modules/deeplearning/cnn_cld_frac.py
index a01a81924acd32a1c3d7b2457187e98a8b1afbaa..417f0d83074d5aa5c514eb455f3bc32a943c3235 100644
--- a/modules/deeplearning/cnn_cld_frac.py
+++ b/modules/deeplearning/cnn_cld_frac.py
@@ -319,7 +319,7 @@ class CNN:
                 for i in range(int(lenx / 2)):
                     cell = grd_k[t, j:j + 2, i:i + 2]
                     if np.sum(np.isnan(cell)) == 0:
-                        cnt = np.sum(cell[t, :, ] == 1.0)
+                        cnt = np.sum(cell)
                         if cnt == 0:
                             grd_down_2x[t, j, i] = 0
                         elif cnt == 1: