diff --git a/modules/icing/util.py b/modules/icing/util.py
index 4a8c12f515df487c4e36df80c7f8ae044b4c845d..9e6a3a273aab3ad3d7e42037889f7ab9e4ceac30 100644
--- a/modules/icing/util.py
+++ b/modules/icing/util.py
@@ -284,7 +284,6 @@ def run_icing_predict_fcn(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_d
     for fidx, fname in enumerate(clvrx_files):
         h5f = h5py.File(fname, 'r')
         dto = clvrx_ds.get_datetime(fname)
-        ts = dto.timestamp()
         clvrx_str_time = dto.strftime('%Y-%m-%d_%H:%M')
 
         data_dct, solzen, satzen, ll, cc = prepare_evaluate(h5f, name_list=train_params, satellite=satellite, domain=domain, offset=8)
@@ -319,8 +318,8 @@ def run_icing_predict_fcn(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_d
                                                                            use_flight_altitude=use_flight_altitude,
                                                                            flight_levels=flight_levels)
             for flvl in flight_levels:
-                preds = preds_day_dct[flvl]
-                probs = probs_day_dct[flvl]
+                preds = (preds_day_dct[flvl]).flatten()
+                probs = (probs_day_dct[flvl]).flatten()
                 fd_preds = preds_2d_dct[flvl]
                 fd_probs = probs_2d_dct[flvl]
                 fd_preds[day_idxs] = preds[day_idxs]
@@ -332,8 +331,8 @@ def run_icing_predict_fcn(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_d
                                                                              use_flight_altitude=use_flight_altitude,
                                                                              flight_levels=flight_levels)
             for flvl in flight_levels:
-                preds = preds_nght_dct[flvl]
-                probs = probs_nght_dct[flvl]
+                preds = (preds_nght_dct[flvl]).flatten()
+                probs = (probs_nght_dct[flvl]).flatten()
                 fd_preds = preds_2d_dct[flvl]
                 fd_probs = probs_2d_dct[flvl]
                 fd_preds[nght_idxs] = preds[nght_idxs]