diff --git a/modules/deeplearning/cnn_cld_frac_mod_res.py b/modules/deeplearning/cnn_cld_frac_mod_res.py
index c8f7062d63fe50c45c8b47c5612fb5664c1c27e8..1d78d8f559c55863a06a5002fb7e5789dd0b6f29 100644
--- a/modules/deeplearning/cnn_cld_frac_mod_res.py
+++ b/modules/deeplearning/cnn_cld_frac_mod_res.py
@@ -65,8 +65,8 @@ params = ['temp_11_0um_nom', 'refl_0_65um_nom', label_param]
 data_params_half = ['temp_11_0um_nom']
 data_params_full = ['refl_0_65um_nom']
 
-# label_idx = params.index(label_param)
-label_idx = 0
+label_idx = params.index(label_param)
+# label_idx = 0
 
 print('data_params_half: ', data_params_half)
 print('data_params_full: ', data_params_full)
@@ -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_label[:, label_idx, :, :]
+        input_label = input_label[:, 0, :, :]
 
         data_norm = []
         for param in data_params_half:
@@ -381,7 +381,7 @@ class SRCNN:
             data_norm.append(avg[:, 0:66, 0:66])
             # data_norm.append(std[:, 0:66, 0:66])
         # ---------------------------------------------------
-        tmp = input_label
+        tmp = input_data[:, label_idx, :, :]
         tmp = tmp.copy()
         tmp = np.where(np.isnan(tmp), 0, tmp)
         if DO_ESPCN: