Fix message classes are not parsed