diff --git a/firebase-dataconnect/src/test/kotlin/com/google/firebase/dataconnect/testutil/property/arbitrary/arbs.kt b/firebase-dataconnect/src/test/kotlin/com/google/firebase/dataconnect/testutil/property/arbitrary/arbs.kt
index 28a7bd45cbd..44c2a5a4720 100644
--- a/firebase-dataconnect/src/test/kotlin/com/google/firebase/dataconnect/testutil/property/arbitrary/arbs.kt
+++ b/firebase-dataconnect/src/test/kotlin/com/google/firebase/dataconnect/testutil/property/arbitrary/arbs.kt
@@ -40,7 +40,6 @@ import io.kotest.property.Arb
 import io.kotest.property.arbitrary.Codepoint
 import io.kotest.property.arbitrary.alphanumeric
 import io.kotest.property.arbitrary.arbitrary
-import io.kotest.property.arbitrary.boolean
 import io.kotest.property.arbitrary.choice
 import io.kotest.property.arbitrary.constant
 import io.kotest.property.arbitrary.enum
@@ -49,7 +48,6 @@ import io.kotest.property.arbitrary.list
 import io.kotest.property.arbitrary.orNull
 import io.kotest.property.arbitrary.string
 import io.mockk.mockk
-import kotlin.reflect.KClass
 import kotlinx.serialization.DeserializationStrategy
 import kotlinx.serialization.SerializationStrategy
 import kotlinx.serialization.modules.SerializersModule
@@ -77,15 +75,6 @@ internal fun DataConnectArb.dataConnectGrpcMetadata(
   )
 }
 
-internal fun DataConnectArb.pathSegmentType(
-  boolean: Arb<Boolean> = Arb.boolean()
-): Arb<KClass<out PathSegment>> = arbitrary {
-  when (boolean.bind()) {
-    true -> PathSegment.Field::class
-    false -> PathSegment.ListIndex::class
-  }
-}
-
 internal fun DataConnectArb.fieldPathSegment(
   string: Arb<String> = string()
 ): Arb<PathSegment.Field> = arbitrary { PathSegment.Field(string.bind()) }