diff --git a/compiler/fir/fir-serialization/src/org/jetbrains/kotlin/fir/serialization/FirContractSerializer.kt b/compiler/fir/fir-serialization/src/org/jetbrains/kotlin/fir/serialization/FirContractSerializer.kt
index a5eeda6371a..2e1a404b597 100644
--- a/compiler/fir/fir-serialization/src/org/jetbrains/kotlin/fir/serialization/FirContractSerializer.kt
+++ b/compiler/fir/fir-serialization/src/org/jetbrains/kotlin/fir/serialization/FirContractSerializer.kt
@@ -6,7 +6,6 @@
 package org.jetbrains.kotlin.fir.serialization
 
 import org.jetbrains.kotlin.contracts.description.*
-import org.jetbrains.kotlin.contracts.description.expressions.*
 import org.jetbrains.kotlin.fir.contracts.FirContractDescription
 import org.jetbrains.kotlin.fir.contracts.description.*
 import org.jetbrains.kotlin.fir.contracts.effects
@@ -58,9 +57,9 @@ class FirContractSerializer {
 
                 is ConeReturnsEffectDeclaration -> {
                     when (effectDeclaration.value) {
-                        ConstantReference.NOT_NULL ->
+                        ConeConstantReference.NOT_NULL ->
                             builder.effectType = ProtoBuf.Effect.EffectType.RETURNS_NOT_NULL
-                        ConstantReference.WILDCARD ->
+                        ConeConstantReference.WILDCARD ->
                             builder.effectType = ProtoBuf.Effect.EffectType.RETURNS_CONSTANT
                         else -> {
                             builder.effectType = ProtoBuf.Effect.EffectType.RETURNS_CONSTANT
@@ -213,4 +212,4 @@ class FirContractSerializer {
                 else -> throw IllegalArgumentException("Unknown constant: $constantReference")
             }
     }
-}
\ No newline at end of file
+}