diff --git a/tests/ai_tests/test_tasks.py b/tests/ai_tests/test_tasks.py
index 35546a45..b07d21bb 100644
--- a/tests/ai_tests/test_tasks.py
+++ b/tests/ai_tests/test_tasks.py
@@ -21,7 +21,7 @@ def test_task_pydantic_result(self):
         assert isinstance(result, Name)
         assert result == Name(first="John", last="Doe")
 
-    @pytest.xfail(reason="Need to revisit dataframe handling")
+    @pytest.mark.xfail(reason="Need to revisit dataframe handling")
     def test_task_dataframe_result(self):
         task = Task(
             'return a dataframe with column "x" that has values 1 and 2 and column "y" that has values 3 and 4',