diff --git a/js/js.tests/test/org/jetbrains/kotlin/js/test/ir/AbstractJsPartialLinkageTestCase.kt b/js/js.tests/test/org/jetbrains/kotlin/js/test/ir/AbstractJsPartialLinkageTestCase.kt
index d01a17a13d9..89ddf7e4b3e 100644
--- a/js/js.tests/test/org/jetbrains/kotlin/js/test/ir/AbstractJsPartialLinkageTestCase.kt
+++ b/js/js.tests/test/org/jetbrains/kotlin/js/test/ir/AbstractJsPartialLinkageTestCase.kt
@@ -77,7 +77,7 @@ abstract class AbstractJsPartialLinkageTestCase(private val compilerType: Compil
     fun runTest(@TestDataFile testPath: String) = PartialLinkageTestUtils.runTest(JsTestConfiguration(testPath))
 
     private fun customizeMainModuleSources(moduleSourceDir: File) {
-        // Add @JsExport annotation to make box function be visible for Node.
+        // Add the @JsExport annotation to make the box function visible to Node.
         moduleSourceDir.walkTopDown().forEach { file ->
             if (file.extension == "kt") {
                 var modified = false
diff --git a/native/native.tests/build.gradle.kts b/native/native.tests/build.gradle.kts
index 215c0ea81dc..9aead14f3d3 100644
--- a/native/native.tests/build.gradle.kts
+++ b/native/native.tests/build.gradle.kts
@@ -28,20 +28,11 @@ dependencies {
     testRuntimeOnly(commonDependency("org.jetbrains.intellij.deps.fastutil:intellij-deps-fastutil"))
 }
 
-val generationRoot = projectDir.resolve("tests-gen")
-
 sourceSets {
     "main" { none() }
     "test" {
         projectDefault()
-        java.srcDirs(generationRoot.name)
-    }
-}
-
-if (kotlinBuildProperties.isInJpsBuildIdeaSync) {
-    apply(plugin = "idea")
-    idea {
-        module.generatedSourceDirs.addAll(listOf(generationRoot))
+        generatedTestDir()
     }
 }