diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 8bd6fb4754fbad7356bf7dfa535ef493262af0c6..b51306f6a9d6f48dd47fcad9adf10f8e9d2a0f2a 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -1198,9 +1198,9 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH',
         for k, w in enumerate(m):
             model_lyrs[k].append(w)
     for lyr in model_lyrs:
-        nda = np.stack(lyr)
+        nda = np.stack(lyr, axis=-1)
         print(nda.shape)
-        avg = np.mean(nda)
+        avg = np.mean(nda, axis=-1)
         print(avg.shape)
         avg_model_weights.append(nda)