diff --git a/docker_stuff/python_stuff/websocket_handler.py b/docker_stuff/python_stuff/websocket_handler.py
index 1940f00..bae225c 100644
--- a/docker_stuff/python_stuff/websocket_handler.py
+++ b/docker_stuff/python_stuff/websocket_handler.py
@@ -168,8 +168,9 @@ def __init__(self, response_data):
                 f"Error decoding JSON message in Extracted Response: {json_error}."
             )
             self.results = ""
-
-        self.message = response_data.get("message", "")
+            
+        if self.event_type == "EVENT":
+            self.message = response_data.get("message", "")
 
 
         self.comment = ""