Fix broken testes