Correction based on comments ( pr )

Signed-off-by: David Saradini <david.saradini@me.com>