diff --git a/modules/deeplearning/cnn_cld_frac_mod_res.py b/modules/deeplearning/cnn_cld_frac_mod_res.py
index 117cb25db91b336319c123b55d8ba402d7cab6d9..c8f7062d63fe50c45c8b47c5612fb5664c1c27e8 100644
--- a/modules/deeplearning/cnn_cld_frac_mod_res.py
+++ b/modules/deeplearning/cnn_cld_frac_mod_res.py
@@ -350,7 +350,7 @@ class SRCNN:
             label_s.append(nda)
         input_data = np.concatenate(data_s)
         input_label = np.concatenate(label_s)
-        input_label = input_data[:, label_idx, :, :]
+        input_label = input_label[:, label_idx, :, :]
 
         data_norm = []
         for param in data_params_half: