fixing balance UI