diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py
index 30fad5c677b0fade9f8cf4ba7206e8e754f3c604..b1daaa11d3ce9c789cc26ba1fb3d313e1e06e87b 100644
--- a/modules/deeplearning/cloud_fraction_fcn_abi.py
+++ b/modules/deeplearning/cloud_fraction_fcn_abi.py
@@ -522,7 +522,7 @@ class SRCNN:
 
         conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_6', kernel_size=KERNEL_SIZE, scale=scale)
 
-        conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, activation=activation, kernel_initializer='he_uniform', padding=padding)(conv_b)
+        conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=KERNEL_SIZE, strides=1, activation=activation, kernel_initializer='he_uniform', padding=padding)(conv_b)
 
         # conv = conv + conv_b
         conv = conv_b