diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index c05cdcffe7bea0d0f8ba0d56f99567b316836107..e9175adcc62ed19a1b1c116bcd2678da533c2141 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -435,7 +435,8 @@ class IcingIntensityNN:
         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