diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index db2fdc69c06af3febf63aaff616074a239ba6d5f..382654bbf4509ac1fb953b3ab0be8dacf555470f 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -719,7 +719,7 @@ class IcingIntensityFCN:
             # optimizer = tfa.optimizers.MovingAverage(optimizer)
             self.ema = tf.train.ExponentialMovingAverage(decay=0.9999)
             self.ema.apply(self.model.trainable_variables)
-            self.ema_trainable_variables = self.ema.average(self.model.trainable_variables)
+            self.ema_trainable_variables = [self.ema.average(var) for var in self.model.trainable_variables]
 
         self.optimizer = optimizer
         self.initial_learning_rate = initial_learning_rate