diff --git a/tests/wallet/did_wallet/test_did.py b/tests/wallet/did_wallet/test_did.py
index 2e43144d093e..75bc67d2d1e6 100644
--- a/tests/wallet/did_wallet/test_did.py
+++ b/tests/wallet/did_wallet/test_did.py
@@ -255,11 +255,9 @@ async def test_creation_from_backup_file(self, self_hostname, three_wallet_nodes
 
         await full_node_api.farm_blocks_to_wallet(1, wallet_0)
 
-        async def get_coins_with_ph():
+        async def get_coins_with_ph() -> bool:
             coins = await full_node_api.full_node.coin_store.get_coin_records_by_puzzle_hash(True, some_ph)
-            if len(coins) == 1:
-                return True
-            return False
+            return len(coins) == 1
 
         await time_out_assert(15, get_coins_with_ph, True)
         await time_out_assert(45, did_wallet_2.get_confirmed_balance, 0)