diff --git a/miner/worker_test.go b/miner/worker_test.go
index fe574b0cde..2df41457a0 100644
--- a/miner/worker_test.go
+++ b/miner/worker_test.go
@@ -427,7 +427,7 @@ func getFakeBorFromConfig(t *testing.T, chainConfig *params.ChainConfig) (consen
 	}, nil).AnyTimes()
 
 	heimdallClientMock := mocks.NewMockIHeimdallClient(ctrl)
-	heimdallClientMock.EXPECT().Close().Times(1)
+	heimdallClientMock.EXPECT().Close().AnyTimes()
 
 	contractMock := bor.NewMockGenesisContract(ctrl)