Fiz alterações nos três códigos concernetes a minha tarefa.
Os casos de testes implementados funcionaram normalmente.
Observação: Fiz os testes e aparentemente está funcionando, porém não tenho certeza se realmente era isso mesmo.