Fix/update balance