diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 232fec740974632d5987d67e00845559b1ab52a9..77d95b9909a291e82e837ca551cc059f74cc07b9 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -598,6 +598,8 @@ class IcingIntensityFCN:
 
         if NOISE_TRAINING:
             conv = tf.keras.layers.GaussianNoise(stddev=NOISE_STDDEV)(input_2d)
+        else:
+            conv = input_2d
 
         conv = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
         print(conv.shape)