diff --git a/libraries/tools/gradle/android-test-fixes/src/main/kotlin/org/jetbrains/kotlin/gradle/test/fixes/android/fixes/debugKeystoreFix.kt b/libraries/tools/gradle/android-test-fixes/src/main/kotlin/org/jetbrains/kotlin/gradle/test/fixes/android/fixes/debugKeystoreFix.kt
index 7e2cfc0749d..1e25a6f104e 100644
--- a/libraries/tools/gradle/android-test-fixes/src/main/kotlin/org/jetbrains/kotlin/gradle/test/fixes/android/fixes/debugKeystoreFix.kt
+++ b/libraries/tools/gradle/android-test-fixes/src/main/kotlin/org/jetbrains/kotlin/gradle/test/fixes/android/fixes/debugKeystoreFix.kt
@@ -5,10 +5,7 @@
 
 package org.jetbrains.kotlin.gradle.test.fixes.android.fixes
 
-import com.android.build.gradle.AppExtension
-import com.android.build.gradle.BaseExtension
-import com.android.build.gradle.FeatureExtension
-import com.android.build.gradle.TestExtension
+import com.android.build.gradle.*
 import org.gradle.api.Action
 import org.gradle.api.Plugin
 import org.gradle.api.Project
@@ -26,7 +23,7 @@ internal fun Project.applyDebugKeystoreFix(
     testFixesProperties: TestFixesProperties
 ) {
     plugins.withId("com.android.application", fix<AppExtension>(testFixesProperties))
-    plugins.withId("com.android.test", fix<TestExtension>(testFixesProperties))
+    plugins.withId("com.android.library", fix<LibraryExtension>(testFixesProperties))
     plugins.withId("com.android.feature", fix<FeatureExtension>(testFixesProperties))
     plugins.withId("com.android.test", fix<TestExtension>(testFixesProperties))
 }