diff --git a/modules/util/augment.py b/modules/util/augment.py
index 9747836f44d3c07e28b12b9b32d5db3e14111935..7c05cb501e61c1a835d1dbc563f4f74ae60f6328 100644
--- a/modules/util/augment.py
+++ b/modules/util/augment.py
@@ -3,17 +3,8 @@ from functools import partial
 import tensorflow as tf
 
 
-def augment_image(
-        brightness_delta=0.05,
-        contrast_factor=[0.7, 1.3],
-        saturation=[0.6, 1.6]):
+def augment_image():
     """ Helper function used for augmentation of images in the dataset.
-      Args:
-        brightness_delta: maximum value for randomly assigning brightness of the image.
-        contrast_factor: list / tuple of minimum and maximum value of factor to set random contrast.
-                          None, if not to be used.
-        saturation: list / tuple of minimum and maximum value of factor to set random saturation.
-                    None, if not to be used.
       Returns:
         tf.data.Dataset mappable function for image augmentation
     """