From ee8cbceda7977cd71833379822e51f87a8785741 Mon Sep 17 00:00:00 2001
From: zebra-lucky <zebra.lucky@gmail.com>
Date: Wed, 1 Oct 2025 00:59:34 +0300
Subject: [PATCH] fix test_taproot_wallet.py test_watchonly_wallet

---
 src/jmclient/__init__.py             |  3 ++-
 test/jmclient/test_taproot_wallet.py | 11 ++++++-----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/jmclient/__init__.py b/src/jmclient/__init__.py
index ad22ade..480b713 100644
--- a/src/jmclient/__init__.py
+++ b/src/jmclient/__init__.py
@@ -27,7 +27,8 @@ from .wallet import (Mnemonic, estimate_tx_fee, WalletError, BaseWallet, ImportW
                      FidelityBondWatchonlyWallet, SegwitWalletFidelityBonds,
                      UTXOManager, WALLET_IMPLEMENTATIONS, compute_tx_locktime,
                      UnknownAddressForLabel, TaprootWallet, FrostWallet,
-                     TaprootWalletFidelityBonds, DKGManager)
+                     TaprootWalletFidelityBonds, DKGManager,
+                     TaprootFidelityBondWatchonlyWallet)
 from .storage import (Argon2Hash, Storage, StorageError, RetryableStorageError,
                       StoragePasswordError, VolatileStorage,
                       DKGStorage, DKGRecoveryStorage)
diff --git a/test/jmclient/test_taproot_wallet.py b/test/jmclient/test_taproot_wallet.py
index 9665b37..bd655f5 100644
--- a/test/jmclient/test_taproot_wallet.py
+++ b/test/jmclient/test_taproot_wallet.py
@@ -21,9 +21,9 @@ from jmclient import (
     load_test_config, jm_single, BaseWallet, BIP32Wallet, VolatileStorage,
     get_network, cryptoengine, WalletError, BIP49Wallet, WalletService,
     TaprootWalletFidelityBonds, create_wallet, open_test_wallet_maybe,
-    open_wallet, FidelityBondMixin, FidelityBondWatchonlyWallet, LegacyWallet,
+    open_wallet, FidelityBondMixin, LegacyWallet,
     wallet_gettimelockaddress, UnknownAddressForLabel, TaprootWallet,
-    get_blockchain_interface_instance)
+    get_blockchain_interface_instance, TaprootFidelityBondWatchonlyWallet)
 from test_blockchaininterface import sync_test_wallet
 from freezegun import freeze_time
 
@@ -998,9 +998,10 @@ class AsyncioTestCase(IsolatedAsyncioTestCase, ParametrizedTestCase):
         watchonly_storage = VolatileStorage()
         entropy = FidelityBondMixin.get_xpub_from_fidelity_bond_master_pub_key(
             master_pub_key).encode()
-        FidelityBondWatchonlyWallet.initialize(watchonly_storage, get_network(),
-            entropy=entropy)
-        watchonly_wallet = FidelityBondWatchonlyWallet(watchonly_storage)
+        TaprootFidelityBondWatchonlyWallet.initialize(
+            watchonly_storage, get_network(), entropy=entropy)
+        watchonly_wallet = TaprootFidelityBondWatchonlyWallet(
+            watchonly_storage)
         await watchonly_wallet.async_init(watchonly_storage)
 
         watchonly_scripts = [