Add getter for entity class in EntityType