diff --git a/test/functional/p2p_add_connections.py b/test/functional/p2p_add_connections.py
index 2d6c03b675..d919ed7bd3 100755
--- a/test/functional/p2p_add_connections.py
+++ b/test/functional/p2p_add_connections.py
@@ -6,7 +6,10 @@
 
 from test_framework.p2p import P2PInterface
 from test_framework.test_framework import BitcoinTestFramework
-from test_framework.util import check_node_connections
+from test_framework.util import (
+    assert_equal,
+    check_node_connections,
+)
 
 class P2PFeelerReceiver(P2PInterface):
     def on_version(self, message):