diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 0c162a633f58a1d9a55197eb27e09f9b150c1227..c435a6a65b505c7c6df9d01404264896ec93b3d5 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -213,7 +213,7 @@ class SRCNN:
         self.test_label_nda = None
 
         # self.n_chans = len(data_params) + 2
-        self.n_chans = 1
+        self.n_chans = 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))