diff --git a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt
index c0e156a70d2..26c2f176444 100644
--- a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt
+++ b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt
@@ -166,7 +166,9 @@ class FirContractResolveTransformer(
             context.storeClass(regularClass)
             regularClass.transformCompanionObject(this, data)
             withTypeParametersOf(regularClass) {
-                regularClass.transformDeclarations(this, data)
+                context.withContainer(regularClass) {
+                    regularClass.transformDeclarations(this, data)
+                }
             }
             return regularClass.compose()
         }