Invoice toggling running balance not correct

When toggling a payment with a related invoice, the running balance shows the total running account balance. However it would be more useful to show the running balance for the invoice which the payment is linked to and was toggled.