From 874c32033121796f7ee74d321449ef5faa204baa Mon Sep 17 00:00:00 2001
From: Tomasz Slabon <tomasz.slabon@keep.network>
Date: Thu, 24 Aug 2023 17:40:07 +0200
Subject: [PATCH] Updated signing done check related unit tests

---
 pkg/tbtc/signing_done_test.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/pkg/tbtc/signing_done_test.go b/pkg/tbtc/signing_done_test.go
index 9dc00f2a3f..792edd6b68 100644
--- a/pkg/tbtc/signing_done_test.go
+++ b/pkg/tbtc/signing_done_test.go
@@ -14,6 +14,7 @@ import (
 	"github.com/keep-network/keep-core/internal/testutils"
 	"github.com/keep-network/keep-core/pkg/chain"
 	"github.com/keep-network/keep-core/pkg/chain/local_v1"
+	"github.com/keep-network/keep-core/pkg/net"
 	"github.com/keep-network/keep-core/pkg/net/local"
 	"github.com/keep-network/keep-core/pkg/operator"
 	"github.com/keep-network/keep-core/pkg/protocol/group"
@@ -326,6 +327,10 @@ func setupSigningDoneCheck(
 		t.Fatal(err)
 	}
 
+	broadcastChannel.SetUnmarshaler(func() net.TaggedUnmarshaler {
+		return &signingDoneMessage{}
+	})
+
 	membershipValidator := group.NewMembershipValidator(
 		&testutils.MockLogger{},
 		operators,