diff --git a/app/client.py b/app/client.py
index 6ea75f6..d68bf50 100644
--- a/app/client.py
+++ b/app/client.py
@@ -156,6 +156,11 @@ async def receive_commands(websocket):
                 command_name = data.get("name")
                 args = data.get("args", [])
 
+                # Check if there is a req ID, if so add it to the args
+                req_id = data.get("request_id", None)
+                if req_id:
+                    args.append(req_id)
+
                 if command_name in command_handlers:
                     print(f"Executing command: {command_name} with args {args}")
                     # Execute the registered async function