From 83b3562be834ac1379743e61df4aa634ea898fea Mon Sep 17 00:00:00 2001
From: JJTech0130 <jjtech@jjtech.dev>
Date: Mon, 24 Jul 2023 16:42:00 -0400
Subject: [PATCH] in case any non-response messages were queued

---
 ids/query.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ids/query.py b/ids/query.py
index 059c97c..0c462eb 100644
--- a/ids/query.py
+++ b/ids/query.py
@@ -53,7 +53,7 @@ def lookup(
         if resp_body is None:
             return False
         resp_body = plistlib.loads(resp_body)
-        return resp_body["U"] == msg_id
+        return resp_body.get('U') == msg_id
 
     # Lambda to check if the response is the one we want
     payload = conn.incoming_queue.wait_pop_find(check_response)