diff --git a/modules/deeplearning/cloud_fraction_fcn.py b/modules/deeplearning/cloud_fraction_fcn.py
index 68ad8d190a0636e684ca5fbf4a69b96152f5c4c3..86b6154a9cb60c4b812a20ee5527e461e62aaf1b 100644
--- a/modules/deeplearning/cloud_fraction_fcn.py
+++ b/modules/deeplearning/cloud_fraction_fcn.py
@@ -795,7 +795,7 @@ 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)
+    grd_b = np.expand_dims(grd_b, 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)