diff --git a/modules/deeplearning/cnn_cld_frac.py b/modules/deeplearning/cnn_cld_frac.py
index 15fd1dd6939d182b984f9ef15c0c9f3543f372da..8b700ec2b17cbd516c6d75d16f13f80c68c7f631 100644
--- a/modules/deeplearning/cnn_cld_frac.py
+++ b/modules/deeplearning/cnn_cld_frac.py
@@ -242,8 +242,8 @@ def get_label_data(grd_k):
     s = a + b + c + d
 
     cat_0 = s <= -3
-    cat_1 = np.logical_and(s > -3, s < 3)
-    cat_2 = s >= 3
+    cat_1 = np.logical_and(s > -3, s < 2)
+    cat_2 = s >= 2
     s[cat_0] = 0
     s[cat_1] = 1
     s[cat_2] = 2