diff --git a/tests/integrations/mcs/tso/api_test.go b/tests/integrations/mcs/tso/api_test.go
index 4cdba3fe3af..fbfe52d6d68 100644
--- a/tests/integrations/mcs/tso/api_test.go
+++ b/tests/integrations/mcs/tso/api_test.go
@@ -200,7 +200,7 @@ func TestForwardOnlyTSONoScheduling(t *testing.T) {
 	re := require.New(t)
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
-	tc, err := tests.NewTestCluster(ctx, 1)
+	tc, err := tests.NewTestClusterWithKeyspaceGroup(ctx, 1)
 	defer tc.Destroy()
 	re.NoError(err)
 	err = tc.RunInitialServers()
diff --git a/tests/integrations/mcs/tso/proxy_test.go b/tests/integrations/mcs/tso/proxy_test.go
index 4b8c7fdba20..152122fe232 100644
--- a/tests/integrations/mcs/tso/proxy_test.go
+++ b/tests/integrations/mcs/tso/proxy_test.go
@@ -62,7 +62,7 @@ func (s *tsoProxyTestSuite) SetupSuite() {
 	var err error
 	s.ctx, s.cancel = context.WithCancel(context.Background())
 	// Create an PD cluster with 1 server
-	s.pdCluster, err = tests.NewTestCluster(s.ctx, 1)
+	s.pdCluster, err = tests.NewTestClusterWithKeyspaceGroup(s.ctx, 1)
 	re.NoError(err)
 	err = s.pdCluster.RunInitialServers()
 	re.NoError(err)