diff --git a/src/crewai/rag/elasticsearch/types.py b/src/crewai/rag/elasticsearch/types.py
index c1f952b5a..9e23c68ab 100644
--- a/src/crewai/rag/elasticsearch/types.py
+++ b/src/crewai/rag/elasticsearch/types.py
@@ -1,7 +1,7 @@
 """Type definitions for Elasticsearch RAG implementation."""
 
-from typing import Any, Protocol, TypedDict, Union, TYPE_CHECKING
-from typing_extensions import NotRequired
+from typing import Any, Protocol, Union, TYPE_CHECKING
+from typing_extensions import NotRequired, TypedDict
 from pydantic import GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
 
diff --git a/tests/rag/elasticsearch/test_config.py b/tests/rag/elasticsearch/test_config.py
index efd9e0979..f1956d5e4 100644
--- a/tests/rag/elasticsearch/test_config.py
+++ b/tests/rag/elasticsearch/test_config.py
@@ -1,7 +1,5 @@
 """Tests for Elasticsearch configuration."""
 
-import pytest
-
 from crewai.rag.elasticsearch.config import ElasticsearchConfig
 
 
diff --git a/tests/rag/elasticsearch/test_factory.py b/tests/rag/elasticsearch/test_factory.py
index e9307ae5c..f0cb6e357 100644
--- a/tests/rag/elasticsearch/test_factory.py
+++ b/tests/rag/elasticsearch/test_factory.py
@@ -1,6 +1,5 @@
 """Tests for Elasticsearch factory."""
 
-import sys
 from unittest.mock import Mock, patch
 
 import pytest