diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index e13d3cf4e4478064daba68bf84a26b028e2f5b48..51f40292490e5dd13d08b82030651008456475b2 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -682,6 +682,7 @@ class IcingIntensityNN:
         self.optimizer.apply_gradients(zip(gradients, self.model.trainable_variables))
         if TRACK_MOVING_AVERAGE:
             self.ema.apply(self.model.trainable_variables)
+            # TODO: This doesn't seem to work
             # for var in self.model.trainable_variables:
             #     var.assign(self.ema.average(var))