Corrigido!
Código posto no fim do arquivo como era anteriormente.