diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 90b1b2224ef64368efc1a89704c5c980a5052588..232fec740974632d5987d67e00845559b1ab52a9 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -317,11 +317,11 @@ class IcingIntensityFCN:
             data_alt_lr = np.copy(data_alt)
             label_lr = np.copy(label)
 
-            data_r1 = np.rot90(data, k=1)
+            data_r1 = np.rot90(data, k=1, axes=(1, 2))
             data_alt_r1 = np.copy(data_alt)
             label_r1 = np.copy(label)
 
-            data_r2 = np.rot90(data, k=1)
+            data_r2 = np.rot90(data, k=1, axes=(1, 2))
             data_alt_r2 = np.copy(data_alt)
             label_r2 = np.copy(label)