diff --git a/build-logic/android-plugins/src/main/kotlin/dev/msfjarvis/aps/gradle/RenameArtifactsPlugin.kt b/build-logic/android-plugins/src/main/kotlin/dev/msfjarvis/aps/gradle/RenameArtifactsPlugin.kt
index 971e3245..dba8353b 100644
--- a/build-logic/android-plugins/src/main/kotlin/dev/msfjarvis/aps/gradle/RenameArtifactsPlugin.kt
+++ b/build-logic/android-plugins/src/main/kotlin/dev/msfjarvis/aps/gradle/RenameArtifactsPlugin.kt
@@ -2,7 +2,7 @@ package dev.msfjarvis.aps.gradle
 
 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 dev.msfjarvis.aps.gradle.artifacts.CollectApksTask
 import dev.msfjarvis.aps.gradle.artifacts.CollectBundleTask
 import org.gradle.api.Plugin
@@ -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"))
           }