diff --git a/tests/patched/money.py b/tests/patched/money.py
index 9592fe66eeb4709fb588a32451e695039a1c86af..8e52e77ac5abb50cc37aefa546f2d950b4a8b6c0 100644
--- a/tests/patched/money.py
+++ b/tests/patched/money.py
@@ -63,6 +63,7 @@ async def patched_get_sources(pubkey):
             )
             balance += amount_in_current_base(listinput[-1])
             a += 1
+        return balance
 
     def listinput_TX(listinput, balance, max_tx):
         a = 0
@@ -78,6 +79,7 @@ async def patched_get_sources(pubkey):
             )
             balance += amount_in_current_base(listinput[-1])
             a += 1
+        return balance
 
     listinput, n = list(), 0
     balance = 0
@@ -123,8 +125,8 @@ async def patched_get_sources(pubkey):
         max_ud = 0
         max_tx = 0
 
-    listinput_UD(listinput, balance, pubkey, max_ud)
-    listinput_TX(listinput, balance, max_tx)
+    balance = listinput_UD(listinput, balance, pubkey, max_ud)
+    balance = listinput_TX(listinput, balance, max_tx)
 
     patched_get_sources.counter += 1
     return listinput, balance