diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index a3159b646ad5e2f21311b548e672469e35e6b50c..781561e15a2e51fbca4813933153b9212013f64e 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -427,7 +427,8 @@ class IcingIntensityFCN:
         dataset = dataset.batch(PROC_BATCH_SIZE)
         dataset = dataset.map(self.data_function, num_parallel_calls=8)
         dataset = dataset.cache()
-        # dataset = dataset.shuffle(PROC_BATCH_BUFFER_SIZE)
+        if DO_AUGMENT:
+            dataset = dataset.shuffle(PROC_BATCH_BUFFER_SIZE)
         dataset = dataset.prefetch(buffer_size=1)
         self.train_dataset = dataset