Correção no mapeamento das classes