diff --git a/toolkit/client/pool_test.go b/toolkit/client/pool_test.go
deleted file mode 100644
index 5ad0d74..0000000
--- a/toolkit/client/pool_test.go
+++ /dev/null
@@ -1,32 +0,0 @@
-package client
-
-import (
-	"testing"
-
-	"github.com/stretchr/testify/require"
-	"github.com/ucan-wg/go-ucan/did/didtest"
-	"github.com/ucan-wg/go-ucan/pkg/command"
-	"github.com/ucan-wg/go-ucan/token/delegation/delegationtest"
-)
-
-func TestFindProof(t *testing.T) {
-	p := NewPool()
-
-	for _, bundle := range delegationtest.AllBundles {
-		p.AddBundle(bundle)
-	}
-
-	require.Equal(t, delegationtest.ProofAliceBob,
-		p.FindProof(delegationtest.NominalCommand, didtest.PersonaBob.DID(), didtest.PersonaAlice.DID()))
-	require.Equal(t, delegationtest.ProofAliceBobCarol,
-		p.FindProof(delegationtest.NominalCommand, didtest.PersonaCarol.DID(), didtest.PersonaAlice.DID()))
-	require.Equal(t, delegationtest.ProofAliceBobCarolDan,
-		p.FindProof(delegationtest.NominalCommand, didtest.PersonaDan.DID(), didtest.PersonaAlice.DID()))
-	require.Equal(t, delegationtest.ProofAliceBobCarolDanErin,
-		p.FindProof(delegationtest.NominalCommand, didtest.PersonaErin.DID(), didtest.PersonaAlice.DID()))
-	require.Equal(t, delegationtest.ProofAliceBobCarolDanErinFrank,
-		p.FindProof(delegationtest.NominalCommand, didtest.PersonaFrank.DID(), didtest.PersonaAlice.DID()))
-
-	// wrong command
-	require.Empty(t, p.FindProof(command.New("foo"), didtest.PersonaBob.DID(), didtest.PersonaAlice.DID()))
-}
diff --git a/toolkit/client/proof_test.go b/toolkit/client/proof_test.go
new file mode 100644
index 0000000..cce0d55
--- /dev/null
+++ b/toolkit/client/proof_test.go
@@ -0,0 +1,33 @@
+package client
+
+import (
+	"iter"
+	"slices"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"github.com/ucan-wg/go-ucan/did/didtest"
+	"github.com/ucan-wg/go-ucan/pkg/command"
+	"github.com/ucan-wg/go-ucan/token/delegation"
+	"github.com/ucan-wg/go-ucan/token/delegation/delegationtest"
+)
+
+func TestFindProof(t *testing.T) {
+	dlgs := func() iter.Seq[*delegation.Bundle] {
+		return slices.Values(delegationtest.AllBundles)
+	}
+
+	require.Equal(t, delegationtest.ProofAliceBob,
+		FindProof(dlgs, delegationtest.NominalCommand, didtest.PersonaBob.DID(), didtest.PersonaAlice.DID()))
+	require.Equal(t, delegationtest.ProofAliceBobCarol,
+		FindProof(dlgs, delegationtest.NominalCommand, didtest.PersonaCarol.DID(), didtest.PersonaAlice.DID()))
+	require.Equal(t, delegationtest.ProofAliceBobCarolDan,
+		FindProof(dlgs, delegationtest.NominalCommand, didtest.PersonaDan.DID(), didtest.PersonaAlice.DID()))
+	require.Equal(t, delegationtest.ProofAliceBobCarolDanErin,
+		FindProof(dlgs, delegationtest.NominalCommand, didtest.PersonaErin.DID(), didtest.PersonaAlice.DID()))
+	require.Equal(t, delegationtest.ProofAliceBobCarolDanErinFrank,
+		FindProof(dlgs, delegationtest.NominalCommand, didtest.PersonaFrank.DID(), didtest.PersonaAlice.DID()))
+
+	// wrong command
+	require.Empty(t, FindProof(dlgs, command.New("foo"), didtest.PersonaBob.DID(), didtest.PersonaAlice.DID()))
+}