diff --git a/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/KotlinLintPlugin.kt b/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/KotlinLintPlugin.kt
index 468b754..262e48e 100644
--- a/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/KotlinLintPlugin.kt
+++ b/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/KotlinLintPlugin.kt
@@ -7,7 +7,6 @@ import io.gitlab.arturbosch.detekt.extensions.DetektExtension
 import io.gitlab.arturbosch.detekt.report.ReportMergeTask
 import org.gradle.api.Plugin
 import org.gradle.api.Project
-import org.gradle.kotlin.dsl.apply
 import org.gradle.kotlin.dsl.getByType
 import org.gradle.kotlin.dsl.register
 import org.gradle.kotlin.dsl.withType
diff --git a/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/KotlinPlugin.kt b/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/KotlinPlugin.kt
index de83446..c7ce81e 100644
--- a/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/KotlinPlugin.kt
+++ b/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/KotlinPlugin.kt
@@ -2,7 +2,6 @@ package io.github.lavenderses.aws_appconfig_openfeature_provider.plugin
 
 import org.gradle.api.Plugin
 import org.gradle.api.Project
-import org.gradle.kotlin.dsl.apply
 
 class KotlinPlugin: Plugin<Project> {
 
diff --git a/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/TestPlugin.kt b/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/TestPlugin.kt
index 843bd9e..8f4a631 100644
--- a/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/TestPlugin.kt
+++ b/build-logic/src/main/kotlin/io/github/lavenderses/aws_appconfig_openfeature_provider/plugin/TestPlugin.kt
@@ -3,13 +3,13 @@ package io.github.lavenderses.aws_appconfig_openfeature_provider.plugin
 import org.gradle.api.Plugin
 import org.gradle.api.Project
 import org.gradle.api.tasks.testing.Test
-import org.gradle.kotlin.dsl.getByName
+import org.gradle.kotlin.dsl.withType
 
 class TestPlugin : Plugin<Project> {
 
     override fun apply(target: Project) {
         with(target) {
-            with(tasks.getByName<Test>("test")) {
+            tasks.withType<Test>() {
                 useJUnitPlatform()
             }