Changed the primary entity class in swagger ui.