diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index ad983722217ab5c79c20b152b4555816c6f1a6bd..5e61e92bdf68e418263c3a1cf513f0f21992d724 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -192,7 +192,7 @@ class SRCNN:
         self.test_data_nda = None
         self.test_label_nda = None
 
-        self.n_chans = len(data_params) + 1
+        self.n_chans = len(data_params) + 2
 
         self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
         # self.X_img = tf.keras.Input(shape=(36, 36, self.n_chans))