From b75a831cf061d83f2cbbc77632338bc8362f72fb Mon Sep 17 00:00:00 2001
From: William Banfield <wbanfield@gmail.com>
Date: Sun, 15 May 2022 22:08:12 -0400
Subject: [PATCH] fix extended commit test

---
 types/block_test.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/types/block_test.go b/types/block_test.go
index b37e087d2..1ebca3ae4 100644
--- a/types/block_test.go
+++ b/types/block_test.go
@@ -580,7 +580,8 @@ func TestVoteSetToExtendedCommit(t *testing.T) {
 			ctx, cancel := context.WithCancel(context.Background())
 			defer cancel()
 
-			voteSet, _, vals := randVoteSet(ctx, t, 3, 1, tmproto.PrecommitType, 10, 1)
+			valSet, vals := randValidatorPrivValSet(ctx, t, 10, 1)
+			voteSet := NewVoteSet("test_chain_id", 3, 1, tmproto.PrecommitType, valSet, testCase.includeExtension)
 			for i := 0; i < len(vals); i++ {
 				pubKey, err := vals[i].GetPubKey(ctx)
 				require.NoError(t, err)