Fix broken tests

Signed-off-by: Vinicius Franco <viniciusf.arantes@gmail.com>
Signed-off-by: Thiago Ribeiro <thiagitosouza@hotmail.com>