diff --git a/keras_wrapper/dataset.py b/keras_wrapper/dataset.py
index 7a9bacc..32328ce 100644
--- a/keras_wrapper/dataset.py
+++ b/keras_wrapper/dataset.py
@@ -3037,7 +3037,7 @@ def preprocessImages(self, path_list, id, set_name, img_size, img_size_crop, use
 
         return data
 
-    def setTrainMean(self, mean_image, id, use_RGB=True, normalization=False):
+    def setTrainMean(self, mean_image, id, normalization=False):
         """
             Loads a pre-calculated training mean image, 'mean_image' can either be:
 
@@ -3046,7 +3046,6 @@ def setTrainMean(self, mean_image, id, use_RGB=True, normalization=False):
             - string with the path to the stored image.
 
         :param mean_image:
-        :param user_RGB: set to False for grayscale images
         :param normalization:
         :param id: identifier of the type of input whose train mean is being introduced.
         """