diff --git a/account.py b/account.py
index a65ac78..a2dea5b 100644
--- a/account.py
+++ b/account.py
@@ -80,8 +80,8 @@ class IncomeStatementCOLGAAP(metaclass=PoolMeta):
     __name__ = 'account_col.income_statement_colgaap'
 
     @classmethod
-    def get_context(cls, records, data):
-        report_context = super(IncomeStatementCOLGAAP, cls).get_context(records, data)
+    def get_context(cls, records, header, data):
+        report_context = super().get_context(records, header, data)
         Company = Pool().get('company.company')
         company = Company(Transaction().context.get('company'))
         result = {}
@@ -131,7 +131,7 @@ class BalanceSheetCOLGAAP(metaclass=PoolMeta):
 
     @classmethod
     def get_context(cls, records, header, data):
-        report_context = super(BalanceSheetCOLGAAP, cls).get_context(records, header, data)
+        report_context = super().get_context(records, header, data)
         Company = Pool().get('company.company')
         company = Company(Transaction().context.get('company'))
         result = {}