adjust entity class