Switch balance to use get_user_balance