diff --git a/cw_monero/lib/api/coins_info.dart b/cw_monero/lib/api/coins_info.dart
index b08dbb1f7..2f09f6e59 100644
--- a/cw_monero/lib/api/coins_info.dart
+++ b/cw_monero/lib/api/coins_info.dart
@@ -38,8 +38,7 @@ Future<int?> getCoinByKeyImage(String keyImage) async {
   final count = await countOfCoins();
   for (int i = 0; i < count; i++) {
     final coin = await getCoin(i);
-    final coinAddress = coin.keyImage;
-    if (keyImage == coinAddress) {
+    if (keyImage == coin.keyImage()) {
       return i;
     }
   }
diff --git a/cw_monero/lib/monero_wallet.dart b/cw_monero/lib/monero_wallet.dart
index eceac205b..e547f5403 100644
--- a/cw_monero/lib/monero_wallet.dart
+++ b/cw_monero/lib/monero_wallet.dart
@@ -879,12 +879,10 @@ abstract class MoneroWalletBase extends WalletBase<MoneroBalance,
   }
 
   void _askForUpdateBalance() {
-    final _ub = _getUnlockedBalance();
-    final _fb =_getFrozenBalance();
-    final unlockedBalance = _ub - _fb;
+    final unlockedBalance = _getUnlockedBalance();
     final fullBalance = monero_wallet.getFullBalance(
-      accountIndex: walletAddresses.account!.id) - _fb;
-    final frozenBalance = _fb;
+      accountIndex: walletAddresses.account!.id);
+    final frozenBalance = _getFrozenBalance();
     if (balance[currency]!.fullBalance != fullBalance ||
         balance[currency]!.unlockedBalance != unlockedBalance ||
         balance[currency]!.frozenBalance != frozenBalance) {