Wrap the entity class