Fixes import for Wallets at 0.6.4