diff --git a/modules/deeplearning/cloud_fraction_fcn.py b/modules/deeplearning/cloud_fraction_fcn.py
index 5160612f6412e02ce833ef6815d8e45de9d8a103..68ad8d190a0636e684ca5fbf4a69b96152f5c4c3 100644
--- a/modules/deeplearning/cloud_fraction_fcn.py
+++ b/modules/deeplearning/cloud_fraction_fcn.py
@@ -795,13 +795,14 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     grd_b = get_grid_values_all(h5f, 'super/refl_0_65um')
     grd_b = np.where(np.isnan(grd_b), 0, grd_b)
     grd_b = grd_b[y_0:y_0+sub_y, x_0:x_0+sub_x]
+    grd_b = grd_b.expand_dims(axis=0)
     lo, hi, std, avg = get_min_max_std(grd_b)
     lo = normalize(lo, 'refl_0_65um_nom', mean_std_dct)
     hi = normalize(hi, 'refl_0_65um_nom', mean_std_dct)
     avg = normalize(avg, 'refl_0_65um_nom', mean_std_dct)
-    lo = lo[slc_y, slc_x]
-    hi = hi[slc_y, slc_x]
-    avg = avg[slc_y, slc_x]
+    lo = lo[0:, slc_y, slc_x]
+    hi = hi[0:, slc_y, slc_x]
+    avg = avg[0, slc_y, slc_x]
 
     grd_c = get_grid_values_all(h5f, 'orig/'+label_param)
     grd_c = np.where(np.isnan(grd_c), 0, grd_c)