diff --git a/tests/extraction_test.py b/tests/extraction_test.py
index ae999a5..04bfaaa 100644
--- a/tests/extraction_test.py
+++ b/tests/extraction_test.py
@@ -1,6 +1,4 @@
-from langchain.pydantic_v1 import BaseModel
-
-from funcchain import chain
+from funcchain import BaseModel, chain
 
 
 class Task(BaseModel):
@@ -15,7 +13,7 @@ def __str__(self) -> str:
 
 def description(task: Task) -> str:
     """
-    CREATE TASK DESCRIPTION: {task}
+    Create a description of the task.
     """
     return chain()