diff --git a/account_financial_report/report/trial_balance.py b/account_financial_report/report/trial_balance.py
index 4f7ce351..1df91a9a 100644
--- a/account_financial_report/report/trial_balance.py
+++ b/account_financial_report/report/trial_balance.py
@@ -330,7 +330,10 @@ class TrialBalanceReport(models.AbstractModel):
         for acc_id, total_data in total_amount.items():
             tmp_list = sorted(
                 total_data.items(),
-                key=lambda x: isinstance(x[1], dict) and x[1]["partner_name"] or x[0],
+                key=lambda x: isinstance(x[0], int)
+                and isinstance(x[1], dict)
+                and x[1]["partner_name"]
+                or x[0],
             )
             total_amount[acc_id] = {}
             for key, value in tmp_list: