Неразрешенная ссылка: sourceSets - Gradle Kotlin DSL в другом файле

Question

В Gradle (Groovy) я использовал это для настройки исходных наборов:

build.gradle * * 1004

apply plugin: 'java'
apply from: 'gradle/test.gradle'

gradle/test.gradle

sourceSets {
    unitTest {
        compileClasspath += main.output
        runtimeClasspath += main.output
    }
}

И тогда я бы использовал исходный набор unitTest для создания задач.

Когда я пытаюсь сделать то же самое в IntelliJ для Gradle KTS:

build.gradle.kts

plugins {
    java
}

apply(from = "gradle/test.gradle.kts")

gradle/test.gradle.kts

sourceSets {
    create("unitTest") {
     // Do configurations
    }
}

Если произошел сбой с ошибкой Unresolved reference: sourceSets.

Как я могу это сделать в Kotlin Gradle?