diff --git a/account_statement_import_online_ponto/models/online_bank_statement_provider_ponto.py b/account_statement_import_online_ponto/models/online_bank_statement_provider_ponto.py
index 6a345af8..ec5adc6f 100644
--- a/account_statement_import_online_ponto/models/online_bank_statement_provider_ponto.py
+++ b/account_statement_import_online_ponto/models/online_bank_statement_provider_ponto.py
@@ -161,9 +161,11 @@ class OnlineBankStatementProvider(models.Model):
         # For each period, create or update statement lines
         for period, statement_lines in grouped_periods.items():
             (date_since, date_until) = period
-            self._create_or_update_statement(
+            statement = self._create_or_update_statement(
                 (statement_lines, {}), date_since, date_until
             )
+            for line in statement.line_ids.filtered(lambda l: not l.partner_id):
+                line.partner_id = line._retrieve_partner()
 
     def _ponto_get_transaction_vals(self, transaction):
         """Translate information from Ponto to statement line vals."""