diff --git a/test/functional/p2p_invalid_messages.py b/test/functional/p2p_invalid_messages.py
index d3284fd8f7..6d4495b471 100755
--- a/test/functional/p2p_invalid_messages.py
+++ b/test/functional/p2p_invalid_messages.py
@@ -16,6 +16,7 @@ from test_framework.messages import (
     MAX_PROTOCOL_MESSAGE_LENGTH,
     msg_getdata,
     msg_headers,
+    msg_headers2,
     msg_inv,
     MSG_TX,
     msg_version,