diff --git a/relay/client/peer_subscription_test.go b/relay/client/peer_subscription_test.go
index 014d8b0b5..0437efa04 100644
--- a/relay/client/peer_subscription_test.go
+++ b/relay/client/peer_subscription_test.go
@@ -25,7 +25,7 @@ func TestWaitToBeOnlineAndSubscribe_Success(t *testing.T) {
 	mockConn := &mockRelayedConn{}
 	logger := logrus.New()
 	logger.SetOutput(&bytes.Buffer{}) // discard log output
-	sub, _ := NewPeersStateSubscription(logrus.NewEntry(logger), mockConn, nil, 0)
+	sub := NewPeersStateSubscription(logrus.NewEntry(logger), mockConn, nil)
 
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
@@ -45,7 +45,7 @@ func TestWaitToBeOnlineAndSubscribe_Timeout(t *testing.T) {
 	mockConn := &mockRelayedConn{}
 	logger := logrus.New()
 	logger.SetOutput(&bytes.Buffer{})
-	sub, _ := NewPeersStateSubscription(logrus.NewEntry(logger), mockConn, nil, 0)
+	sub := NewPeersStateSubscription(logrus.NewEntry(logger), mockConn, nil)
 
 	ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
 	defer cancel()
@@ -60,7 +60,7 @@ func TestWaitToBeOnlineAndSubscribe_Duplicate(t *testing.T) {
 	mockConn := &mockRelayedConn{}
 	logger := logrus.New()
 	logger.SetOutput(&bytes.Buffer{})
-	sub, _ := NewPeersStateSubscription(logrus.NewEntry(logger), mockConn, nil, 0)
+	sub := NewPeersStateSubscription(logrus.NewEntry(logger), mockConn, nil)
 
 	ctx := context.Background()
 	go func() {
@@ -78,7 +78,7 @@ func TestUnsubscribeStateChange(t *testing.T) {
 	mockConn := &mockRelayedConn{}
 	logger := logrus.New()
 	logger.SetOutput(&bytes.Buffer{})
-	sub, _ := NewPeersStateSubscription(logrus.NewEntry(logger), mockConn, nil, 0)
+	sub := NewPeersStateSubscription(logrus.NewEntry(logger), mockConn, nil)
 
 	doneChan := make(chan struct{})
 	go func() {