Add sanity check for balance