Added message_fields_schema so can get entity key.