diff --git a/src/crewai/task.py b/src/crewai/task.py
index b4e3bc758..7ef9bd651 100644
--- a/src/crewai/task.py
+++ b/src/crewai/task.py
@@ -7,7 +7,18 @@ from concurrent.futures import Future
 from copy import copy
 from hashlib import md5
 from pathlib import Path
-from typing import Any, ClassVar, Dict, List, Optional, Set, Tuple, Type, Union
+from typing import (
+    Any,
+    Callable,
+    ClassVar,
+    Dict,
+    List,
+    Optional,
+    Set,
+    Tuple,
+    Type,
+    Union,
+)
 
 from opentelemetry.trace import Span
 from pydantic import (