diff --git a/api/rpk-api/firm/current_firm/routes.py b/api/rpk-api/firm/current_firm/routes.py index b4a803d..3f5c506 100644 --- a/api/rpk-api/firm/current_firm/routes.py +++ b/api/rpk-api/firm/current_firm/routes.py @@ -31,7 +31,7 @@ async def create(schema: CurrentFirmSchemaCreate, reg=Depends(get_uninitialized_ primary_color=schema.primary_color, secondary_color=schema.secondary_color, )) - return CurrentFirmSchemaRead.from_model(document) + return CurrentFirmSchemaRead.from_model_and_entities(document, EntityRead.from_model(corp), EntityRead.from_model(owner_entity)) @current_firm_router.put("/", response_description=f"Current Firm record updated") async def update(schema: CurrentFirmSchemaUpdate, reg=Depends(get_authed_tenant_registry)) -> CurrentFirmSchemaRead: