diff --git a/pkg/port/beego/controllers/partner_info_controller.go b/pkg/port/beego/controllers/partner_info_controller.go
index a5b1296..a8f86c5 100644
--- a/pkg/port/beego/controllers/partner_info_controller.go
+++ b/pkg/port/beego/controllers/partner_info_controller.go
@@ -60,6 +60,7 @@ func (c *PartnerInfoController) CreatePartnerInfo() {
 		c.ResponseError(e)
 		return
 	}
+	companyId := c.GetUserCompany()
 	cmd := partnerInfoCmd.CreatePartnerInfoCommand{
 		PartnerName:     param.PartnerName,
 		Account:         param.Account,
@@ -73,6 +74,7 @@ func (c *PartnerInfoController) CreatePartnerInfo() {
 				Telephone: param.Phone,
 			},
 		},
+		CompanyId: companyId,
 	}
 	if len(param.Area) > 0 {
 		cmd.RegionInfo = &domain.RegionInfo{
@@ -116,6 +118,7 @@ func (c *PartnerInfoController) UpdatePartnerInfo() {
 		c.ResponseError(e)
 		return
 	}
+	companyId := c.GetUserCompany()
 	cmd := partnerInfoCmd.UpdatePartnerInfoCommand{
 		Id:              param.ID,
 		Status:          param.State,
@@ -130,6 +133,7 @@ func (c *PartnerInfoController) UpdatePartnerInfo() {
 				Telephone: param.Phone,
 			},
 		},
+		CompanyId: companyId,
 	}
 	serve := partnerInfoService.NewPartnerInfoService(nil)
 	err = serve.UpdatePartnerInfo(&cmd)