Corrected tests

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