diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index b4b91fdd4dc13d398555f9444ee02ffd3a4984b3..c2539a9ae4a59911fa1ffd4f5b9476cd0c899b64 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -219,7 +219,7 @@ 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(5))
+        self.inputs.append(tf.keras.Input(None, None, 5))
 
         self.flight_level = 0