diff --git a/demo/kotlin-demo/src/test/kotlin/com/alibaba/demo/basic/DemoMockTest.kt b/demo/kotlin-demo/src/test/kotlin/com/alibaba/demo/basic/DemoMockTest.kt
index f6638b4..1df5a36 100644
--- a/demo/kotlin-demo/src/test/kotlin/com/alibaba/demo/basic/DemoMockTest.kt
+++ b/demo/kotlin-demo/src/test/kotlin/com/alibaba/demo/basic/DemoMockTest.kt
@@ -109,7 +109,7 @@ internal class DemoMockTest {
     }
 
     @Test
-    fun should_get_test_case_name() {
+    fun should_set_mock_context() {
         MOCK_CONTEXT["case"] = "special_case"
         // synchronous
         assertEquals("mock_special", demoMock.callerOne())
diff --git a/docs/en-us/doc/use-mock.md b/docs/en-us/doc/use-mock.md
index 9f26a07..cf2dedc 100644
--- a/docs/en-us/doc/use-mock.md
+++ b/docs/en-us/doc/use-mock.md
@@ -159,7 +159,7 @@ private Data mockDemo() {
 }
 ```
 
-For complete code examples, see the `should_get_source_method_name()` and `should_get_test_case_name()` test cases in the `java-demo` and `kotlin-demo` sample projects.
+For complete code examples, see the `should_get_source_method_name()` and `should_set_mock_context()` test cases in the `java-demo` and `kotlin-demo` sample projects.
 
 ### 3. Verify the sequence and parameters of the mock method being invoked
 
diff --git a/docs/zh-cn/doc/use-mock.md b/docs/zh-cn/doc/use-mock.md
index 517097c..d7875d9 100644
--- a/docs/zh-cn/doc/use-mock.md
+++ b/docs/zh-cn/doc/use-mock.md
@@ -159,7 +159,7 @@ private Data mockDemo() {
 }
 ```
 
-完整代码示例见`java-demo`和`kotlin-demo`示例项目中的`should_get_source_method_name()`和`should_get_test_case_name()`测试用例。
+完整代码示例见`java-demo`和`kotlin-demo`示例项目中的`should_get_source_method_name()`和`should_set_mock_context()`测试用例。
 
 ### 3. 验证Mock方法被调用的顺序和参数