fix balance UI