Fixed the wallet bug