diff --git a/baseline-profile/src/main/java/app/lawnchair/baseline/Constants.kt b/baseline-profile/src/main/java/app/lawnchair/baseline/Constants.kt
index 90ba5cef2b..d5c6cef48c 100644
--- a/baseline-profile/src/main/java/app/lawnchair/baseline/Constants.kt
+++ b/baseline-profile/src/main/java/app/lawnchair/baseline/Constants.kt
@@ -1,5 +1,11 @@
 package app.lawnchair.baseline
 
+import androidx.test.platform.app.InstrumentationRegistry
+
 object Constants {
-    val PACKAGE_NAME = "app.lawnchair"
+    val PACKAGE_NAME: String
+        get() = InstrumentationRegistry.getArguments()
+            .getString("androidx.benchmark.targetPackageName")
+            ?: InstrumentationRegistry.getInstrumentation().targetContext.packageName
+                .removeSuffix(".baseline")
 }
diff --git a/build.gradle b/build.gradle
index f242466e63..05e0f8296d 100644
--- a/build.gradle
+++ b/build.gradle
@@ -354,6 +354,12 @@ android {
     }
 }
 
+baselineProfile {
+    saveInSrc = true
+    mergeIntoMain = true
+    automaticGenerationDuringBuild = false
+}
+
 composeCompiler {
     stabilityConfigurationFiles.addAll(
         layout.projectDirectory.file("compose_compiler_config.conf"),