diff --git a/modules/deeplearning/cloud_fraction_fcn_viirs.py b/modules/deeplearning/cloud_fraction_fcn_viirs.py
index 1c05f540cab7936b2cb414dc7188e377c9bbd083..217f6e4673822d6d02bdbbcdff4ab70ab1eb4cf9 100644
--- a/modules/deeplearning/cloud_fraction_fcn_viirs.py
+++ b/modules/deeplearning/cloud_fraction_fcn_viirs.py
@@ -70,14 +70,13 @@ print('data_params_full: ', data_params_full)
 print('label_param: ', label_param)
 
 KERNEL_SIZE = 3  # target size: (128, 128)
-N_X = N_Y = 1
 X_LEN = Y_LEN = 128
 
 if KERNEL_SIZE == 3:
-    slc_x = slice(1, int((N_X*X_LEN)/2) + 3)
-    slc_y = slice(1, int((N_Y*Y_LEN)/2) + 3)
-    x_128 = slice(4, N_X*X_LEN + 4)
-    y_128 = slice(4, N_Y*Y_LEN + 4)
+    slc_x = slice(1, int(X_LEN/2) + 3)
+    slc_y = slice(1, int(Y_LEN/2) + 3)
+    x_128 = slice(4, X_LEN + 4)
+    y_128 = slice(4, Y_LEN + 4)
 # elif KERNEL_SIZE == 5: These no longer apply here
 #     slc_x = slice(3, 135)
 #     slc_y = slice(3, 135)