Small fix to the wallet