diff --git a/relayer/relayer_test.go b/relayer/relayer_test.go
index ec8ab3ea..aab60835 100644
--- a/relayer/relayer_test.go
+++ b/relayer/relayer_test.go
@@ -70,6 +70,7 @@ func (s *RouteTestSuite) TestReceiveMessageFails() {
 
 func (s *RouteTestSuite) TestAvoidWriteWithoutProposals() {
 	s.mockRelayedChain.EXPECT().ReceiveMessage(gomock.Any()).Return(nil, nil)
+	s.mockRelayedChain.EXPECT().DomainID().Return(uint8(1))
 	chains := make(map[uint8]RelayedChain)
 	chains[1] = s.mockRelayedChain
 	relayer := NewRelayer(
@@ -105,6 +106,7 @@ func (s *RouteTestSuite) TestWritesToChain() {
 	props[0] = prop
 	s.mockRelayedChain.EXPECT().ReceiveMessage(gomock.Any()).Return(prop, nil)
 	s.mockRelayedChain.EXPECT().Write(props).Return(nil)
+	s.mockRelayedChain.EXPECT().DomainID().Return(uint8(1)).Times(1)
 	chains := make(map[uint8]RelayedChain)
 	chains[1] = s.mockRelayedChain
 	relayer := NewRelayer(