diff --git a/modules/deeplearning/cloud_fraction_fcn.py b/modules/deeplearning/cloud_fraction_fcn.py
index a924feea07597b6db44e1922eb6eafc02bfc123c..2bccf2da38e02b7913d574b64e660b9a8175e1c9 100644
--- a/modules/deeplearning/cloud_fraction_fcn.py
+++ b/modules/deeplearning/cloud_fraction_fcn.py
@@ -787,6 +787,9 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     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 = np.squeeze(lo)
+    hi = np.squeeze(hi)
+    avg = np.squeeze(avg)
 
     grd_c = get_grid_values_all(h5f, 'orig/'+label_param)
     grd_c = np.where(np.isnan(grd_c), 0, grd_c)