Ok! Encontrei um código no próprio projeto que me ajudou a fazer o que eu queria.
Agora, este trecho de código está mais simples e segue o padrão usado no projeto.