diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py
index e1ac70b2219f87ae5fc437840e0496815e919863..942c991b3de7f0e4e939052c4eab8af406cca519 100644
--- a/modules/deeplearning/cloud_opd_fcn_abi.py
+++ b/modules/deeplearning/cloud_opd_fcn_abi.py
@@ -290,7 +290,7 @@ class SRCNN:
         # self.n_chans = len(data_params_half) + len(data_params_full) + 1
         self.n_chans = 5
 
-        self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
+        self.X_img = tf.keras.Input(shape=(None, None, self.n_chans + 1))
 
         self.inputs.append(self.X_img)