diff --git a/plugins/kotlin/idea/tests/testData/formatter/ContextParameterSpacesInFunctionType.after.kt b/plugins/kotlin/idea/tests/testData/formatter/ContextParameterSpacesInFunctionType.after.kt
index eab99a4c672e..577795bfc39e 100644
--- a/plugins/kotlin/idea/tests/testData/formatter/ContextParameterSpacesInFunctionType.after.kt
+++ b/plugins/kotlin/idea/tests/testData/formatter/ContextParameterSpacesInFunctionType.after.kt
@@ -1,3 +1,3 @@
-fun <R> analyze(action: context(KaSession) (Int, String, Float) -> R): R = TODO()
+fun <R> analyze(action: context(Int, String, Float, String) (Int, String, Float) -> R): R = TODO()
 
 // SET_TRUE: SPACE_AFTER_CONTEXT_PARAMETER_LIST_IN_FUNCTION_TYPE
diff --git a/plugins/kotlin/idea/tests/testData/formatter/ContextParameterSpacesInFunctionType.kt b/plugins/kotlin/idea/tests/testData/formatter/ContextParameterSpacesInFunctionType.kt
index 989b0d79dc1b..5a27a922ff75 100644
--- a/plugins/kotlin/idea/tests/testData/formatter/ContextParameterSpacesInFunctionType.kt
+++ b/plugins/kotlin/idea/tests/testData/formatter/ContextParameterSpacesInFunctionType.kt
@@ -1,3 +1,3 @@
-fun <R> analyze(action: context(KaSession)  (  Int,   String,     Float )  ->     R):  R = TODO()
+fun <R> analyze(action: context(      Int,    String,Float   , String   )  (  Int,   String,     Float )  ->     R):  R = TODO()
 
 // SET_TRUE: SPACE_AFTER_CONTEXT_PARAMETER_LIST_IN_FUNCTION_TYPE