diff --git a/utils/queues.py b/utils/queues.py
index 3f520ea..bbae082 100644
--- a/utils/queues.py
+++ b/utils/queues.py
@@ -10,7 +10,7 @@ class QueuesException(Exception):
 class TasksHandlerMixin:
     def poll(self):
         while True:
-            response: tasks_pb2.TakeResponse = self.stub.Take(tasks_pb2.TakeRequest(self.queue_name))
+            response: tasks_pb2.TakeResponse = self.stub.Take(tasks_pb2.TakeRequest(queue=self.queue_name))
             task: tasks_pb2.Task = response.task
             if not task:
                 time.sleep(0.2)
@@ -21,7 +21,7 @@ class TasksHandlerMixin:
                 print(f'Error processing message id={task["id"]}, payload={task["payload"]}, exc={exc}')
                 continue
             try:
-                self.stub.Finish(tasks_pb2.FinishRequest(task.id))
+                self.stub.Finish(tasks_pb2.FinishRequest(id=task.id))
             except:
                 print(f'Failed to finish task id={task.id}')