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