Fixed build errors caused by class name changes.

Signed-off-by: Vinicius Dias de Oliveira Gardelli <gardelli123@gmail.com>