Corrigindo import errado em um teste