diff --git a/build-logic/convention/src/main/kotlin/AndroidApplicationTestOptionsConventionPlugin.kt b/build-logic/convention/src/main/kotlin/AndroidApplicationTestOptionsConventionPlugin.kt
index 6f17e6ddb..3ec95b01a 100644
--- a/build-logic/convention/src/main/kotlin/AndroidApplicationTestOptionsConventionPlugin.kt
+++ b/build-logic/convention/src/main/kotlin/AndroidApplicationTestOptionsConventionPlugin.kt
@@ -1,9 +1,3 @@
-import com.android.build.api.dsl.ApplicationExtension
-import com.google.samples.apps.nowinandroid.configureTestOptions
-import org.gradle.api.Plugin
-import org.gradle.api.Project
-import org.gradle.kotlin.dsl.getByType
-
 /*
  * Copyright 2024 The Android Open Source Project
  *
@@ -20,6 +14,12 @@ import org.gradle.kotlin.dsl.getByType
  * limitations under the License.
  */
 
+import com.android.build.api.dsl.ApplicationExtension
+import com.google.samples.apps.nowinandroid.configureTestOptions
+import org.gradle.api.Plugin
+import org.gradle.api.Project
+import org.gradle.kotlin.dsl.configure
+
 class AndroidApplicationTestOptionsConventionPlugin : Plugin<Project> {
     override fun apply(target: Project) {
         with(target) {
diff --git a/build-logic/convention/src/main/kotlin/AndroidLibraryTestOptionsConventionPlugin.kt b/build-logic/convention/src/main/kotlin/AndroidLibraryTestOptionsConventionPlugin.kt
index ac2bedab4..5703248bb 100644
--- a/build-logic/convention/src/main/kotlin/AndroidLibraryTestOptionsConventionPlugin.kt
+++ b/build-logic/convention/src/main/kotlin/AndroidLibraryTestOptionsConventionPlugin.kt
@@ -1,10 +1,3 @@
-
-import com.android.build.api.dsl.LibraryExtension
-import com.google.samples.apps.nowinandroid.configureTestOptions
-import org.gradle.api.Plugin
-import org.gradle.api.Project
-import org.gradle.kotlin.dsl.getByType
-
 /*
  * Copyright 2024 The Android Open Source Project
  *
@@ -21,6 +14,12 @@ import org.gradle.kotlin.dsl.getByType
  * limitations under the License.
  */
 
+import com.android.build.api.dsl.LibraryExtension
+import com.google.samples.apps.nowinandroid.configureTestOptions
+import org.gradle.api.Plugin
+import org.gradle.api.Project
+import org.gradle.kotlin.dsl.configure
+
 class AndroidLibraryTestOptionsConventionPlugin : Plugin<Project> {
     override fun apply(target: Project) {
         with(target) {