diff --git a/p2p/discover/portal_protocol_test.go b/p2p/discover/portal_protocol_test.go
index 8b4732fd51..efc49ae7c3 100644
--- a/p2p/discover/portal_protocol_test.go
+++ b/p2p/discover/portal_protocol_test.go
@@ -114,14 +114,14 @@ func TestPortalWireProtocolUdp(t *testing.T) {
 	assert.NoError(t, err)
 	time.Sleep(12 * time.Second)
 
-	node1.putCacheNodeId(node2.localNode.Node())
-	node1.putCacheNodeId(node3.localNode.Node())
+	node1.cacheNode(node2.localNode.Node())
+	node1.cacheNode(node3.localNode.Node())
 
-	node2.putCacheNodeId(node1.localNode.Node())
-	node2.putCacheNodeId(node3.localNode.Node())
+	node2.cacheNode(node1.localNode.Node())
+	node2.cacheNode(node3.localNode.Node())
 
-	node3.putCacheNodeId(node1.localNode.Node())
-	node3.putCacheNodeId(node2.localNode.Node())
+	node3.cacheNode(node1.localNode.Node())
+	node3.cacheNode(node2.localNode.Node())
 
 	udpAddrStr1 := fmt.Sprintf("%s:%d", node1.localNode.Node().IP(), node1.localNode.Node().UDP())
 	udpAddrStr2 := fmt.Sprintf("%s:%d", node2.localNode.Node().IP(), node2.localNode.Node().UDP())