fix class names

Signed-off-by: Eduardo Stuart <stuart.eduardo@gmail.com>