diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 9bef8299d7e0e2db943691663b7d3f1ef393fb53..f9ef3d6a1d6e317e38ca4d6f0bdfaae89b06dfe0 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -199,7 +199,7 @@ class IcingIntensityNN:
             data.append(nda)
         data = np.stack(data)
         data = data.astype(np.float32)
-        data = np.transpose(data, axes=(1, 0))
+        data = np.transpose(data, axes=(1, 2, 3, 0))
 
         label = self.h5f['icing_intensity'][nd_idxs]
         label = label.astype(np.int32)