Atualizando imports nos testes