diff --git a/modules/util/augment.py b/modules/util/augment.py
index 68c39c61e8d90e2771fa5527bcf8735a8bd0137d..790d4f1bd6d0a2aeb70d1d222d2396224e6cea89 100644
--- a/modules/util/augment.py
+++ b/modules/util/augment.py
@@ -45,7 +45,7 @@ def augment_image():
     return augment_fn
 
 
-def augment_image_3arg():
+def augment_icing():
     """ Helper function used for augmentation of images in the dataset.
       Returns:
         tf.data.Dataset mappable function for image augmentation