diff --git a/tests/code_context_test.py b/tests/code_context_test.py
index 5c046f5ca..ce384b159 100644
--- a/tests/code_context_test.py
+++ b/tests/code_context_test.py
@@ -213,20 +213,20 @@ def func_4(string):
     mock_session_context.config.auto_context_tokens = 8000
 
     code_message = await code_context.get_code_message(0, prompt="prompt")
-    assert mock_session_context.llm_api_handler.spice.count_tokens(code_message, "gpt-4", is_message=True) == 95  # Code
+    assert mock_session_context.llm_api_handler.spice.count_tokens(code_message, "gpt-4", is_message=True) == 85  # Code
     assert (
         code_message
         == """\
 Code Files:
 
-file_1.py (search-result, user-included)
+file_1.py
 1:def func_1(x, y):
 2:    return x + y
 3:
 4:def func_2():
 5:    return 3
 
-file_2.py (search-result)
+file_2.py
 1:def func_3(a, b, c):
 2:    return a * b ** c
 3:
diff --git a/tests/code_file_manager_test.py b/tests/code_file_manager_test.py
index 8873b0da0..659fa4e8b 100644
--- a/tests/code_file_manager_test.py
+++ b/tests/code_file_manager_test.py
@@ -51,7 +51,7 @@ async def test_partial_files(mocker, mock_session_context):
         """\
             Code Files:
 
-            dir/file.txt (user-included)
+            dir/file.txt
             1:I am a file
             ...
             3:third