diff --git a/addons/stock/models/res_company.py b/addons/stock/models/res_company.py
index 58c61be714f..b78080e023c 100644
--- a/addons/stock/models/res_company.py
+++ b/addons/stock/models/res_company.py
@@ -28,11 +28,11 @@ class Company(models.Model):
             location.sudo().write({'company_id': company.id})
             company.write({'internal_transit_location_id': location.id})
 
-        warehouses = self.env['stock.warehouse'].search([('partner_id', '=', self.partner_id.id)])
-        warehouses.mapped('partner_id').with_context(force_company=self.id).write({
-            'property_stock_customer': location.id,
-            'property_stock_supplier': location.id,
-        })
+            warehouses = self.env['stock.warehouse'].search([('partner_id', '=', location.partner_id.id)])
+            warehouses.mapped('partner_id').with_context(force_company=location.id).write({
+                'property_stock_customer': location.id,
+                'property_stock_supplier': location.id,
+            })
 
     @api.model
     def create(self, vals):