diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 05e2284d3c936701365b85a798cea3a08d92c69d..5b31a587332c1d6054b631e419760219daaa6e6c 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -220,7 +220,8 @@ class IcingIntensityFCN:
         self.X_img = tf.keras.Input(shape=(None, None, n_chans))
 
         self.inputs.append(self.X_img)
-        self.inputs.append(tf.keras.Input(shape=(None, None, 5)))
+        #self.inputs.append(tf.keras.Input(shape=(None, None, 5)))
+        self.inputs.append(tf.keras.Input(shape=(5)))
 
         self.flight_level = 0
 
@@ -358,8 +359,6 @@ class IcingIntensityFCN:
         nda[np.logical_and(nda >= 8000, nda < 15000)] = 4
 
         nda = tf.one_hot(nda, 5).numpy()
-        nda = np.expand_dims(nda, axis=1)
-        nda = np.expand_dims(nda, axis=1)
 
         return nda