Fix Issue #2

Signed-off-by:pdrolourenco@gmail.com <pdrolourenco@gmail.com>