diff --git a/account_statement_ext/statement.py b/account_statement_ext/statement.py
index 7022c366..80edbfcc 100644
--- a/account_statement_ext/statement.py
+++ b/account_statement_ext/statement.py
@@ -223,12 +223,17 @@ class AccountBankSatement(Model):
         move of period_id to the statement line
         """
         for statement in self.browse(cr, uid, ids, context=context):
+            # statement.company_id is a related store=True that for some
+            # reason doesn't work in YAML tests. As a workaround, I unwind it
+            # to statement.journal_id.company_id here.
             if (statement.period_id and
-                    statement.company_id.id != statement.period_id.company_id.id):
+                    statement.journal_id.company_id.id !=
+                    statement.period_id.company_id.id):
                 return False
             for line in statement.line_ids:
                 if (line.period_id and
-                        statement.company_id.id != line.period_id.company_id.id):
+                        statement.journal_id.company_id.id
+                        != line.period_id.company_id.id):
                     return False
         return True