diff --git a/tests/unittest/test_find_line_number_of_relevant_line_in_file.py b/tests/unittest/test_find_line_number_of_relevant_line_in_file.py
index f9d143c6..fcb028ca 100644
--- a/tests/unittest/test_find_line_number_of_relevant_line_in_file.py
+++ b/tests/unittest/test_find_line_number_of_relevant_line_in_file.py
@@ -1,6 +1,6 @@
 
 # Generated by CodiumAI
-from pr_agent.git_providers.types import FilePatchInfo
+from pr_agent.algo.types import FilePatchInfo
 from pr_agent.algo.utils import find_line_number_of_relevant_line_in_file
 
 import pytest