Fix balance display issue