1 files changed, 6 insertions, 3 deletions
diff --git a/build-logic/android-plugins/src/main/kotlin/app/passwordstore/gradle/RenameArtifactsPlugin.kt b/build-logic/android-plugins/src/main/kotlin/app/passwordstore/gradle/RenameArtifactsPlugin.kt
index 04eb57e3..6807b6ef 100644
--- a/build-logic/android-plugins/src/main/kotlin/app/passwordstore/gradle/RenameArtifactsPlugin.kt
+++ b/build-logic/android-plugins/src/main/kotlin/app/passwordstore/gradle/RenameArtifactsPlugin.kt
@@ -4,7 +4,7 @@ import app.passwordstore.gradle.artifacts.CollectApksTask
 import app.passwordstore.gradle.artifacts.CollectBundleTask
 import com.android.build.api.artifact.SingleArtifact
 import com.android.build.api.variant.ApplicationAndroidComponentsExtension
-import com.android.build.gradle.internal.dsl.BaseAppModuleExtension
+import com.android.build.api.variant.VariantOutputConfiguration
 import org.gradle.api.Plugin
 import org.gradle.api.Project
 import org.gradle.kotlin.dsl.getByType
@@ -15,7 +15,6 @@ class RenameArtifactsPlugin : Plugin<Project> {
 
   override fun apply(project: Project) {
     project.pluginManager.withPlugin("com.android.application") {
-      val android = project.extensions.getByType<BaseAppModuleExtension>()
       project.extensions.getByType<ApplicationAndroidComponentsExtension>().run {
         onVariants { variant ->
           project.tasks.register<CollectApksTask>("collect${variant.name.capitalize()}Apks") {
@@ -25,8 +24,12 @@ class RenameArtifactsPlugin : Plugin<Project> {
             outputDirectory.set(project.layout.projectDirectory.dir("outputs"))
           }
           project.tasks.register<CollectBundleTask>("collect${variant.name.capitalize()}Bundle") {
+            val mainOutput =
+              variant.outputs.single {
+                it.outputType == VariantOutputConfiguration.OutputType.SINGLE
+              }
             variantName.set(variant.name)
-            versionName.set(android.defaultConfig.versionName)
+            versionName.set(mainOutput.versionName)
             bundleFile.set(variant.artifacts.get(SingleArtifact.BUNDLE))
             outputDirectory.set(project.layout.projectDirectory.dir("outputs"))
           }