diff --git a/firebase-ai/firebase-ai.gradle.kts b/firebase-ai/firebase-ai.gradle.kts
index e65cb8674a7..2aa3a16b6fb 100644
--- a/firebase-ai/firebase-ai.gradle.kts
+++ b/firebase-ai/firebase-ai.gradle.kts
@@ -66,10 +66,7 @@ android {
     targetSdk = targetSdkVersion
     baseline = file("lint-baseline.xml")
   }
-  sourceSets {
-    // getByName("test").java.srcDirs("src/testUtil")
-    getByName("androidTest") { kotlin.srcDirs("src/testUtil") }
-  }
+  sourceSets { getByName("test").java.srcDirs("src/testUtil") }
 }
 
 kotlin {