diff --git a/cognee/tests/test_repo_processor.py b/cognee/tests/test_repo_processor.py
index 4de102da6..fc3c26b05 100644
--- a/cognee/tests/test_repo_processor.py
+++ b/cognee/tests/test_repo_processor.py
@@ -1,7 +1,7 @@
 import os
 import shutil
 import tempfile
-from cognee.tasks.repo_processor.code_graph_repo import get_source_code_files
+from cognee.tasks.repo_processor.get_repo_file_dependencies import get_source_code_files
 
 def test_get_source_code_files_excludes_common_dirs_and_files():
     # Create a temporary test directory