diff --git a/modules/deeplearning/cloud_fraction_fcn.py b/modules/deeplearning/cloud_fraction_fcn.py
index 21afd77780ebbd6703512f98a709298c1c0dd3bd..ccb394608cf16d3dad4d23c6101696a90440aba0 100644
--- a/modules/deeplearning/cloud_fraction_fcn.py
+++ b/modules/deeplearning/cloud_fraction_fcn.py
@@ -820,7 +820,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     var_names = ['cloud_fraction', 'temp_11_0um', 'refl_0_65um']
     dims = ['num_params', 'y', 'x']
 
-    da = xr.DataArray(np.column_stack([cld_frac_out, bt, refl_avg]), dims=dims)
+    da = xr.DataArray(np.stack([cld_frac_out, bt, refl_avg], axis=0), dims=dims)
     da.assign_coords({
         'lat': (['y', 'x'], lats),
         'lon': (['y', 'x'], lons)