Fixes class name

Signed-off-by: Lucas Martins <lucasilvamartins@gmail.com>