From f7cf87142fc3c5f4ccd3cbad7dbd0db11530b026 Mon Sep 17 00:00:00 2001
From: caelansar <31852257+caelansar@users.noreply.github.com>
Date: Sat, 19 Apr 2025 01:20:17 -0500
Subject: [PATCH] Fix callback deletion in UDP transport

---
 dns/transport/udp.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dns/transport/udp.go b/dns/transport/udp.go
index ec17c71b..e2ad9db7 100644
--- a/dns/transport/udp.go
+++ b/dns/transport/udp.go
@@ -117,7 +117,7 @@ func (t *UDPTransport) exchange(ctx context.Context, message *mDNS.Msg) (*mDNS.M
 	conn.access.Unlock()
 	defer func() {
 		conn.access.Lock()
-		delete(conn.callbacks, messageId)
+		delete(conn.callbacks, exMessage.Id)
 		conn.access.Unlock()
 	}()
 	rawMessage, err := exMessage.PackBuffer(buffer.FreeBytes())