diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 560e9f61dc68481084c040fb7f1dff62399cd491..db1f7aad21d1b9ef4ac024588753d90bcef0f7f6 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -221,7 +221,7 @@ class SRCNN:
             tmp = normalize(tmp, param, mean_std_dct, add_noise=add_noise, noise_scale=noise_scale)
             data_norm.append(tmp)
         # --------
-        tmp = input_data[:, 2, 3:131:2, 3:131:2]
+        tmp = input_data[:, label_idx, 3:131:2, 3:131:2]
         tmp = resample_2d_linear(x_64, y_64, tmp, t, s)
         #tmp = normalize(tmp, 'temp_11_0um_nom', mean_std_dct)
         data_norm.append(tmp)