diff --git a/Dockerfile b/Dockerfile
index d905ed4..ba79138 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,6 +8,6 @@ RUN pip install -r requirements.txt
 
 COPY . .
 
-ENV PYTHONUNBUFFERED 1
+ENV PYTHONUNBUFFERED=1
 
 ENTRYPOINT ["python", "main.py"]
diff --git a/app/routers/take.py b/app/routers/take.py
index 1f6c3a9..53a1ecb 100644
--- a/app/routers/take.py
+++ b/app/routers/take.py
@@ -23,9 +23,9 @@ class Response(pydantic.BaseModel):
 async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response:
     try:
         task = await tasks.take_task(queue)
+        if not task:
+            return Response(task=None)
+        return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload))
     except Exception as e:
-        print('GOT ERROR', e)
+        print("GOT ERROR", e)
         return Response(task=None)
-    if not task:
-        return Response(task=None)
-    return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload))