diff --git a/api/onnx_web/worker/pool.py b/api/onnx_web/worker/pool.py
index 803d4773..9b8925f3 100644
--- a/api/onnx_web/worker/pool.py
+++ b/api/onnx_web/worker/pool.py
@@ -256,7 +256,7 @@ class DevicePoolExecutor:
         )
 
         device = self.devices[device_idx].device
-        self.pending[device].put((fn, args, kwargs))
+        self.pending[device].put((key, fn, args, kwargs))
 
     def status(self) -> List[Tuple[str, int, bool, bool]]:
         history = [
diff --git a/api/onnx_web/worker/worker.py b/api/onnx_web/worker/worker.py
index d6e2d0c9..6755cd1f 100644
--- a/api/onnx_web/worker/worker.py
+++ b/api/onnx_web/worker/worker.py
@@ -18,11 +18,8 @@ def worker_main(context: WorkerContext, server: ServerContext):
     logger.info("checking in from worker, %s", get_available_providers())
 
     while True:
-        job = context.pending.get()
-        logger.info("got job: %s", job)
-
-        fn, args, kwargs = job
-        name = args[3][0]
+        name, fn, args, kwargs = context.pending.get()
+        logger.info("worker for %s got job: %s", context.device.device, name)
 
         try:
             context.job = name  # TODO: hax