Ótimos pontos! Criei esse código pensando no meu caso de uso e acabei esquecendo de generalizar ele. Vou atualizar o código com suas dicas!