diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py
index 4aa3b729359bd253851ddebda3ad778b5d3438af..a5094858dfb41409fdd54c6fe6cb86e14fa0c518 100644
--- a/modules/deeplearning/cloud_opd_fcn_abi.py
+++ b/modules/deeplearning/cloud_opd_fcn_abi.py
@@ -299,7 +299,9 @@ class SRCNN:
         # self.n_chans = len(data_params_half) + len(data_params_full) + 1
         self.n_chans = 6
 
-        self.X_img = tf.keras.Input(shape=(None, None, self.n_chans + 2))
+        # Testing/Evaluation mode
+        # self.X_img = tf.keras.Input(shape=(None, None, self.n_chans + 2))
+        self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
 
         self.inputs.append(self.X_img)