diff --git a/docs/concepts/tasks.mdx b/docs/concepts/tasks.mdx
index 79164ec1..36fab98f 100644
--- a/docs/concepts/tasks.mdx
+++ b/docs/concepts/tasks.mdx
@@ -289,7 +289,7 @@ If you exclude `completion_tools`, agents may be unable to complete the task or
 
 Here are some examples:
 
-```
+```python
 # Generate both success and failure tools (default behavior, equivalent to `completion_tools=None`)
 task = cf.Task(
     objective="Write a poem about AI",