fixing balance UI 3