diff --git a/compiler/testData/codegen/box/when/enumOptimization/differentEnumClasses.kt b/compiler/testData/codegen/box/when/enumOptimization/differentEnumClasses.kt
index b8e893eeb4a..189128144ae 100644
--- a/compiler/testData/codegen/box/when/enumOptimization/differentEnumClasses.kt
+++ b/compiler/testData/codegen/box/when/enumOptimization/differentEnumClasses.kt
@@ -1,4 +1,6 @@
+// !LANGUAGE: -ProhibitComparisonOfIncompatibleEnums
 // IGNORE_BACKEND_FIR: JVM_IR
+
 enum class A {
     O, K
 }
diff --git a/compiler/testData/codegen/box/when/enumOptimization/differentEnumClasses2.kt b/compiler/testData/codegen/box/when/enumOptimization/differentEnumClasses2.kt
index 61c0a47012d..7574cb99e69 100644
--- a/compiler/testData/codegen/box/when/enumOptimization/differentEnumClasses2.kt
+++ b/compiler/testData/codegen/box/when/enumOptimization/differentEnumClasses2.kt
@@ -1,4 +1,6 @@
+// !LANGUAGE: -ProhibitComparisonOfIncompatibleEnums
 // IGNORE_BACKEND_FIR: JVM_IR
+
 enum class A {
     OK
 }