diff --git a/compiler/tests-spec/testData/diagnostics/notLinked/contracts/declarations/contractBuilder/common/neg/5.kt b/compiler/tests-spec/testData/diagnostics/notLinked/contracts/declarations/contractBuilder/common/neg/5.kt
index e9439ea10a9..9235332e4e9 100644
--- a/compiler/tests-spec/testData/diagnostics/notLinked/contracts/declarations/contractBuilder/common/neg/5.kt
+++ b/compiler/tests-spec/testData/diagnostics/notLinked/contracts/declarations/contractBuilder/common/neg/5.kt
@@ -18,7 +18,7 @@ fun case_1(): Boolean {
 }
 
 fun case_2(): Boolean {
-    contract { returnsNotNull() implies (return return return true) }
+    <!ERROR_IN_CONTRACT_DESCRIPTION!>contract<!> { returnsNotNull() implies (return return return true) }
     return true
 }
 
diff --git a/compiler/tests-spec/testData/diagnostics/notLinked/contracts/declarations/contractBuilder/common/neg/5.txt b/compiler/tests-spec/testData/diagnostics/notLinked/contracts/declarations/contractBuilder/common/neg/5.txt
index 3dc2b09c64b..c75b746f078 100644
--- a/compiler/tests-spec/testData/diagnostics/notLinked/contracts/declarations/contractBuilder/common/neg/5.txt
+++ b/compiler/tests-spec/testData/diagnostics/notLinked/contracts/declarations/contractBuilder/common/neg/5.txt
@@ -3,6 +3,5 @@ package
 public fun case_1(): kotlin.Boolean
 public fun case_2(): kotlin.Boolean
 public fun case_3(): kotlin.Boolean
-public fun case_4(): kotlin.Boolean
-public fun case_6(): kotlin.Boolean?
-public fun case_7(/*0*/ value_1: kotlin.Boolean): kotlin.Boolean?
+public fun case_5(): kotlin.Boolean?
+public fun case_6(/*0*/ value_1: kotlin.Boolean): kotlin.Boolean?