diff --git a/bob/ip/pytorch_extractor/LightCNN9.py b/bob/ip/pytorch_extractor/LightCNN9.py
index 5c0e39ba7828aa017873a433462ae890f3bed1b8..66d9c781344a8f7f25a6ecaa28550d3c2d20d300 100644
--- a/bob/ip/pytorch_extractor/LightCNN9.py
+++ b/bob/ip/pytorch_extractor/LightCNN9.py
@@ -65,19 +65,18 @@ class LightCNN9Extractor(Extractor):
 
     Parameters
     ----------
-    image : 3D :py:class:`numpy.ndarray` (floats)
-      The image to extract the features from. Its size must be 3x128x128
+    image : 2D :py:class:`numpy.ndarray` (floats)
+      The grayscale image to extract the features from. Its size must be 128x128
 
     Returns
     -------
-    feature : 2D :py:class:`numpy.ndarray` (floats)
+    feature : :py:class:`numpy.ndarray` (floats)
       The extracted features as a 1d array of size 320 
     
     """
   
     # torchvision.transforms expect a numpy array of size HxWxC
     input_image = numpy.expand_dims(image, axis=2)
-    
     input_image = self.to_tensor(input_image)
     input_image = self.norm(input_image)
     input_image = input_image.unsqueeze(0)