diff --git a/test_clustering_integration.py b/test_clustering_integration.py
index d4feb03e..22f136b2 100755
--- a/test_clustering_integration.py
+++ b/test_clustering_integration.py
@@ -18,9 +18,10 @@
     create_component_id_map,
     normalize_component_ids_by_lookup,
 )
+from codewiki.src.be.dependency_analyzer.utils.logging_config import setup_logging
 
 # Configure logging to capture warnings
-logging.basicConfig(level=logging.WARNING, format='%(message)s')
+setup_logging(level=logging.WARNING, format='%(message)s')
 
 
 class TestResults:
diff --git a/test_clustering_proof.py b/test_clustering_proof.py
index 9dfbecc2..da1b0da9 100644
--- a/test_clustering_proof.py
+++ b/test_clustering_proof.py
@@ -1,8 +1,10 @@
 #!/usr/bin/env python3
-import os, sys, logging
-logging.basicConfig(level=logging.INFO, format='[%(levelname)s] %(message)s', force=True)
+import os, sys
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 
+from codewiki.src.be.dependency_analyzer.utils.logging_config import setup_logging
+setup_logging()
+
 from dotenv import load_dotenv
 load_dotenv('.env.local')