diff --git a/docs/docs/usage-guide/additional_configurations.md b/docs/docs/usage-guide/additional_configurations.md
index 8d8e75dc..121d77b6 100644
--- a/docs/docs/usage-guide/additional_configurations.md
+++ b/docs/docs/usage-guide/additional_configurations.md
@@ -66,7 +66,7 @@ By default, around any change in your PR, git patch provides three lines of cont
 For the `review`, `describe`, `ask` and `add_docs` tools, if the token budget allows, PR-Agent tries to increase the number of lines of context, via the parameter:
 ```
 [config]
-patch_extra_lines_before=6
+patch_extra_lines_before=4
 patch_extra_lines_after=2
 ```
 
diff --git a/tests/unittest/test_extend_patch.py b/tests/unittest/test_extend_patch.py
index cb2b3c9c..7737ee8d 100644
--- a/tests/unittest/test_extend_patch.py
+++ b/tests/unittest/test_extend_patch.py
@@ -1,5 +1,6 @@
 import pytest
 from pr_agent.algo.git_patch_processing import extend_patch
+from pr_agent.algo.pr_processing import pr_generate_extended_diff
 from pr_agent.algo.token_handler import TokenHandler
 
 
@@ -60,7 +61,7 @@ class TestExtendPatch:
         assert actual_output == expected_output
 
 
-class PRProcessingTest:
+class TestExtendedPatchMoreLines:
     class File:
         def __init__(self, base_file, patch, filename):
             self.base_file = base_file