diff --git a/docs/examples/custom_storage_knowledge_source_example.py b/docs/examples/custom_storage_knowledge_source_example.py
index 58ec84fe0..6bd6a5673 100644
--- a/docs/examples/custom_storage_knowledge_source_example.py
+++ b/docs/examples/custom_storage_knowledge_source_example.py
@@ -1,10 +1,14 @@
 """Example of using a custom storage with CrewAI."""
 
+from pathlib import Path
+
 import chromadb
 from chromadb.config import Settings
 
 from crewai import Agent, Crew, Task
-from crewai.knowledge.source.custom_storage_knowledge_source import CustomStorageKnowledgeSource
+from crewai.knowledge.source.custom_storage_knowledge_source import (
+    CustomStorageKnowledgeSource,
+)
 from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
 
 
diff --git a/tests/knowledge/custom_storage_knowledge_source_test.py b/tests/knowledge/custom_storage_knowledge_source_test.py
index 73935631e..a7c8dab08 100644
--- a/tests/knowledge/custom_storage_knowledge_source_test.py
+++ b/tests/knowledge/custom_storage_knowledge_source_test.py
@@ -1,13 +1,14 @@
 """Test CustomStorageKnowledgeSource functionality."""
 
-import os
 from pathlib import Path
 from unittest.mock import patch
 
 import pytest
 
 from crewai.knowledge.knowledge import Knowledge
-from crewai.knowledge.source.custom_storage_knowledge_source import CustomStorageKnowledgeSource
+from crewai.knowledge.source.custom_storage_knowledge_source import (
+    CustomStorageKnowledgeSource,
+)
 from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage