diff --git a/modules/deeplearning/cloud_fraction_fcn_viirs.py b/modules/deeplearning/cloud_fraction_fcn_viirs.py
index 29722007a88d0516db84b7142605a607a652a3a8..7441cb1e6e068156d58efd34b47450e5f68ca273 100644
--- a/modules/deeplearning/cloud_fraction_fcn_viirs.py
+++ b/modules/deeplearning/cloud_fraction_fcn_viirs.py
@@ -314,11 +314,11 @@ class SRCNN:
         data_norm = []
         for param in data_params_half:
             # If next 2 uncommented, take out get_grid_cell_mean
-            # idx = params.index(param)
-            # tmp = input_data[:, idx, :, :]
-            idx = params_i.index(param)
-            tmp = input_label[:, idx, :, :]
-            tmp = get_grid_cell_mean(tmp)
+            idx = params.index(param)
+            tmp = input_data[:, idx, :, :]
+            # idx = params_i.index(param)
+            # tmp = input_label[:, idx, :, :]
+            # tmp = get_grid_cell_mean(tmp)
             tmp = tmp[:, slc_y, slc_x]
             tmp = normalize(tmp, param, mean_std_dct)
             data_norm.append(tmp)
@@ -334,7 +334,7 @@ class SRCNN:
 
             data_norm.append(lo[:, slc_y, slc_x])
             data_norm.append(hi[:, slc_y, slc_x])
-            data_norm.append(std[:, slc_y, slc_x])
+            data_norm.append(avg[:, slc_y, slc_x])
         # ---------------------------------------------------
         # If next uncommented, take out get_grid_cell_mean
         # tmp = input_data[:, label_idx, :, :]