diff --git a/payroll.py b/payroll.py
index bf3468f..4c81935 100644
--- a/payroll.py
+++ b/payroll.py
@@ -996,11 +996,12 @@ class PayrollRecompute(Wizard):
 
     def transition_do_recompute(self):
         Payroll = Pool().get('staff.payroll')
+        wages_dict = Payroll.create_cache_wage_types()
         ids = Transaction().context['active_ids']
         for payroll in Payroll.browse(ids):
             if payroll.state != 'draft' or not payroll.lines:
                 continue
-            payroll.recompute_lines()
+            payroll.recompute_lines(wages_dict)
         return 'end'