Little modification into the code

Signed-off-by: Jean-François Erdelyi <jferdelyi@gmail.com>