Correcting tests

Signed-off-by: Michael Granados <michaelgranados@gmail.com>