Fix current balance

Fix current balance