diff --git a/tests/plugins/replace_payload.py b/tests/plugins/replace_payload.py
index 122899653..e515d8d45 100755
--- a/tests/plugins/replace_payload.py
+++ b/tests/plugins/replace_payload.py
@@ -32,7 +32,7 @@ def on_htlc_accepted(htlc, onion, plugin, **kwargs):
 
 
 @plugin.method('setpayload')
-def setpayload(plugin, payload: bool):
+def setpayload(plugin, payload: bytes):
     plugin.replace_payload = payload
     return {}