diff --git a/cognee/api/v1/cognify/code_graph_pipeline.py b/cognee/api/v1/cognify/code_graph_pipeline.py
index ee83cfa66..42ff86e86 100644
--- a/cognee/api/v1/cognify/code_graph_pipeline.py
+++ b/cognee/api/v1/cognify/code_graph_pipeline.py
@@ -32,7 +32,9 @@ from cognee.tasks.storage import add_data_points
 
 from cognee.base_config import get_base_config
 from cognee.shared.data_models import MonitoringTool
-if MonitoringTool.LANGFUSE:
+
+monitoring = get_base_config().monitoring_tool
+if monitoring == MonitoringTool.LANGFUSE:
     from langfuse.decorators import observe
 
 from cognee.tasks.summarization import summarize_code