diff --git a/partner_statement/wizard/statement_common.py b/partner_statement/wizard/statement_common.py
index 2a89f505..d1072d52 100644
--- a/partner_statement/wizard/statement_common.py
+++ b/partner_statement/wizard/statement_common.py
@@ -14,7 +14,7 @@ class StatementCommon(models.AbstractModel):
     def _get_company(self):
         return (
             self.env["res.company"].browse(self.env.context.get("force_company"))
-            or self.env.user.company_id
+            or self.env.company
         )
 
     name = fields.Char()