diff --git a/idea/testData/resolve/partialBodyResolve/ContractFromSource.completion b/idea/testData/resolve/partialBodyResolve/ContractFromSource.completion
index 4710faa24e3..8bb5ea2a3b4 100644
--- a/idea/testData/resolve/partialBodyResolve/ContractFromSource.completion
+++ b/idea/testData/resolve/partialBodyResolve/ContractFromSource.completion
@@ -1,11 +1,11 @@
 Resolve target: value-parameter x: kotlin.Any? smart-cast to kotlin.String
 ----------------------------------------------
 // COMPILER_ARGUMENTS: -XXLanguage:+AllowContractsForCustomFunctions -XXLanguage:+UseReturnsEffect
-@file:Suppress("INVISIBLE_MEMBER")
 package test
 
-import kotlin.internal.contracts.*
+import kotlin.contracts.*
 
+@UseExperimental(ExperimentalContracts::class)
 fun myRequire(x: Boolean) {
     contract {
         returns() implies x
@@ -19,4 +19,4 @@ fun testContractFromSource(x: Any?, y: Any?) {
     <caret>x.length
 
     /* STATEMENT DELETED: myRequire(x is Int) */
-}
+}
\ No newline at end of file
diff --git a/idea/testData/resolve/partialBodyResolve/ContractFromSource.dump b/idea/testData/resolve/partialBodyResolve/ContractFromSource.dump
index 938100b2f93..d5446bcc433 100644
--- a/idea/testData/resolve/partialBodyResolve/ContractFromSource.dump
+++ b/idea/testData/resolve/partialBodyResolve/ContractFromSource.dump
@@ -1,11 +1,11 @@
 Resolve target: value-parameter x: kotlin.Any? smart-cast to kotlin.String
 ----------------------------------------------
 // COMPILER_ARGUMENTS: -XXLanguage:+AllowContractsForCustomFunctions -XXLanguage:+UseReturnsEffect
-@file:Suppress("INVISIBLE_MEMBER")
 package test
 
-import kotlin.internal.contracts.*
+import kotlin.contracts.*
 
+@UseExperimental(ExperimentalContracts::class)
 fun myRequire(x: Boolean) {
     contract {
         returns() implies x
@@ -19,4 +19,4 @@ fun testContractFromSource(x: Any?, y: Any?) {
     <caret>x.length
 
     /* STATEMENT DELETED: myRequire(x is Int) */
-}
+}
\ No newline at end of file
diff --git a/idea/testData/resolve/partialBodyResolve/ContractFromSource.kt b/idea/testData/resolve/partialBodyResolve/ContractFromSource.kt
index 700ffac9d0a..efe7b83135c 100644
--- a/idea/testData/resolve/partialBodyResolve/ContractFromSource.kt
+++ b/idea/testData/resolve/partialBodyResolve/ContractFromSource.kt
@@ -1,9 +1,9 @@
 // COMPILER_ARGUMENTS: -XXLanguage:+AllowContractsForCustomFunctions -XXLanguage:+UseReturnsEffect
-@file:Suppress("INVISIBLE_MEMBER")
 package test
 
-import kotlin.internal.contracts.*
+import kotlin.contracts.*
 
+@UseExperimental(ExperimentalContracts::class)
 fun myRequire(x: Boolean) {
     contract {
         returns() implies x