diff --git a/test/integration/submitqueue/core/consumer/consumer_test.go b/test/integration/submitqueue/core/consumer/consumer_test.go
index fc558d9f..fd0ed984 100644
--- a/test/integration/submitqueue/core/consumer/consumer_test.go
+++ b/test/integration/submitqueue/core/consumer/consumer_test.go
@@ -126,13 +126,14 @@ func (s *ConsumerIntegrationSuite) newConsumer(t *testing.T, q extqueue.Queue, t
 			Name:  topicName,
 			Queue: q,
 			Subscription: extqueue.SubscriptionConfig{
-				SubscriberName:         "test-worker",
-				ConsumerGroup:          consumerGroup,
-				PollIntervalMs:         100,
-				BatchSize:              10,
-				VisibilityTimeoutMs:    60000,
-				LeaseRenewalIntervalMs: 2000,
-				LeaseDurationMs:        5000,
+				SubscriberName:               "test-worker",
+				ConsumerGroup:                consumerGroup,
+				PollIntervalMs:               100,
+				PartitionDiscoveryIntervalMs: 100,
+				BatchSize:                    10,
+				VisibilityTimeoutMs:          60000,
+				LeaseRenewalIntervalMs:       2000,
+				LeaseDurationMs:              5000,
 				Retry: extqueue.RetryConfig{
 					MaxAttempts:       3,
 					InitialBackoffMs:  1000,