Use correct argument

Signed-off-by: Hector Acosta <hector.acosta@gmail.com>