diff --git a/cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/extract_summary.py b/cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/extract_summary.py
index 3ccc8f817..efedb0f02 100644
--- a/cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/extract_summary.py
+++ b/cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/extract_summary.py
@@ -5,7 +5,7 @@ from baml_py import ClientRegistry
 from cognee.shared.logging_utils import get_logger
 from cognee.shared.data_models import SummarizedCode
 from cognee.infrastructure.llm.structured_output_framework.baml.baml_client.async_client import b
-from cognee.infrastructure.llm.structured_output_framework.baml_src.config import get_llm_config
+from cognee.infrastructure.llm.config import get_llm_config
 
 config = get_llm_config()