diff --git a/src/crewai/utilities/exceptions/agent_execution_timeout_error.py b/src/crewai/utilities/exceptions/agent_execution_timeout_error.py
index f2c0fbba5..8435dbc6f 100644
--- a/src/crewai/utilities/exceptions/agent_execution_timeout_error.py
+++ b/src/crewai/utilities/exceptions/agent_execution_timeout_error.py
@@ -1,7 +1,7 @@
 class AgentExecutionTimeoutError(Exception):
     """Exception raised when an agent execution exceeds the maximum allowed time."""
     
-    def __init__(self, max_execution_time: int, message: str = None):
+    def __init__(self, max_execution_time: int, message: str | None = None):
         self.max_execution_time = max_execution_time
         self.message = message or f"Agent execution exceeded maximum allowed time of {max_execution_time} seconds"
         super().__init__(self.message)
diff --git a/tests/test_timeout/test_agent_timeout.py b/tests/test_timeout/test_agent_timeout.py
index 26cfba5ab..71ab86d9c 100644
--- a/tests/test_timeout/test_agent_timeout.py
+++ b/tests/test_timeout/test_agent_timeout.py
@@ -1,6 +1,7 @@
-import time
 import pytest
-from unittest.mock import patch, MagicMock
+import time
+from unittest.mock import MagicMock, patch
+
 from crewai import Agent, Task
 
 def test_agent_max_execution_time():