Use EntityType instead of the entity class