diff --git a/pkg/querier/tail/querier_test.go b/pkg/querier/tail/querier_test.go
index b0f5d113b0..8818b16be8 100644
--- a/pkg/querier/tail/querier_test.go
+++ b/pkg/querier/tail/querier_test.go
@@ -60,6 +60,7 @@ func TestQuerier_Tail_QueryTimeoutConfigFlag(t *testing.T) {
 	}
 	ingester.On("Tail", mock.Anything, &request).Return(clients, nil)
 	ingester.On("TailersCount", mock.Anything).Return([]uint32{0}, nil)
+	ingester.On("TailDisconnectedIngesters", mock.Anything, mock.Anything, mock.Anything).Return(map[string]logproto.Querier_TailClient{}, nil).Maybe()
 
 	// Setup log selector mock
 	logSelector := newMockTailLogSelector()
@@ -153,6 +154,7 @@ func TestQuerier_concurrentTailLimits(t *testing.T) {
 			}
 			ingester.On("Tail", mock.Anything, &request).Return(clients, testData.ingesterError)
 			ingester.On("TailersCount", mock.Anything).Return([]uint32{testData.tailersCount}, testData.ingesterError)
+			ingester.On("TailDisconnectedIngesters", mock.Anything, mock.Anything, mock.Anything).Return(map[string]logproto.Querier_TailClient{}, nil).Maybe()
 
 			// Setup log selector mock
 			logSelector := newMockTailLogSelector()