diff --git a/build-logic/src/main/kotlin/dev/msfjarvis/claw/gradle/deps.ext.kt b/build-logic/src/main/kotlin/dev/msfjarvis/claw/gradle/deps.ext.kt
index 6f499479..cc6c645a 100644
--- a/build-logic/src/main/kotlin/dev/msfjarvis/claw/gradle/deps.ext.kt
+++ b/build-logic/src/main/kotlin/dev/msfjarvis/claw/gradle/deps.ext.kt
@@ -16,7 +16,7 @@ import org.gradle.kotlin.dsl.exclude
 /** Extension function to configure JUnit5 dependencies with the Truth assertion library. */
 fun DependencyHandlerScope.addTestDependencies(project: Project) {
   val libs = project.extensions.getByName("libs") as LibrariesForLibs
-  arrayOf("test", "androidTest")
+  arrayOf("test", "androidTest", "screenshotTest")
     .filter { sourceSet -> project.configurations.findByName("${sourceSet}Implementation") != null }
     .forEach { sourceSet ->
       addProvider("${sourceSet}Implementation", libs.junit.jupiter.api)
diff --git a/common/build.gradle.kts b/common/build.gradle.kts
index ff39be14..4f7dbf0b 100644
--- a/common/build.gradle.kts
+++ b/common/build.gradle.kts
@@ -7,6 +7,7 @@
 @file:Suppress("UnstableApiUsage")
 
 import com.android.build.api.variant.HasUnitTestBuilder
+import dev.msfjarvis.claw.gradle.addTestDependencies
 
 plugins {
   id("dev.msfjarvis.claw.android-library")
@@ -64,4 +65,6 @@ dependencies {
   compileOnly(libs.androidx.compose.ui.tooling.preview)
 
   runtimeOnly(libs.androidx.compose.ui.tooling)
+
+  addTestDependencies(project)
 }