diff --git a/ads/aqua/extension/ui_websocket_handler.py b/ads/aqua/extension/ui_websocket_handler.py
index 701a9b023..fa3a4e254 100644
--- a/ads/aqua/extension/ui_websocket_handler.py
+++ b/ads/aqua/extension/ui_websocket_handler.py
@@ -112,6 +112,7 @@ def on_message_processed(self, future: concurrent.futures.Future):
             )
             raise e
         finally:
+            self.future_message_map.pop(future)
             # Send the response back to the client on the event thread
             IOLoop.current().run_sync(lambda: self.write_message(response.to_json()))