diff --git a/tests/plugins/onionmessage-reply.py b/tests/plugins/onionmessage-reply.py
index 66ad0611c..2caa8b3b9 100755
--- a/tests/plugins/onionmessage-reply.py
+++ b/tests/plugins/onionmessage-reply.py
@@ -2,7 +2,7 @@
 """
 This plugin is used to test the `onion_message` hook.
 """
-from lightning import Plugin
+from pyln.client import Plugin
 
 plugin = Plugin()