diff --git a/compiler/fir/modularized-tests/tests/org/jetbrains/kotlin/fir/NonFirResolveModularizedTotalKotlinTest.kt b/compiler/fir/modularized-tests/tests/org/jetbrains/kotlin/fir/NonFirResolveModularizedTotalKotlinTest.kt
index 1d548aa48f1..91f6b9c01be 100644
--- a/compiler/fir/modularized-tests/tests/org/jetbrains/kotlin/fir/NonFirResolveModularizedTotalKotlinTest.kt
+++ b/compiler/fir/modularized-tests/tests/org/jetbrains/kotlin/fir/NonFirResolveModularizedTotalKotlinTest.kt
@@ -69,14 +69,15 @@ class NonFirResolveModularizedTotalKotlinTest : AbstractModularizedTest() {
 
         val configuration = createDefaultConfiguration(moduleData)
 
-        if (USE_NI) {
-            configuration.languageVersionSettings =
-                LanguageVersionSettingsImpl(
-                    LanguageVersion.KOTLIN_1_4, ApiVersion.KOTLIN_1_3, specificFeatures = mapOf(
-                        LanguageFeature.NewInference to LanguageFeature.State.ENABLED
-                    )
+
+        configuration.languageVersionSettings =
+            LanguageVersionSettingsImpl(
+                configuration.languageVersionSettings.languageVersion,
+                configuration.languageVersionSettings.apiVersion,
+                specificFeatures = mapOf(
+                    LanguageFeature.NewInference to if (USE_NI) LanguageFeature.State.ENABLED else LanguageFeature.State.DISABLED
                 )
-        }
+            )
 
         System.getProperty("fir.bench.oldfe.jvm_target")?.let {
             configuration.put(JVMConfigurationKeys.JVM_TARGET, JvmTarget.fromString(it) ?: error("Unknown JvmTarget"))