Adicionei o import que faltava para funcionar