diff --git a/tests/tso/client_test.go b/tests/tso/client_test.go
index 1650dfc60279..465390f163f8 100644
--- a/tests/tso/client_test.go
+++ b/tests/tso/client_test.go
@@ -82,7 +82,7 @@ func (suite *tsoClientTestSuite) SetupSuite() {
 		re.NoError(err)
 	} else {
 		suite.tsoServer, suite.tsoServerCleanup = mcs.StartSingleTSOTestServer(suite.ctx, re, backendEndpoints)
-		suite.client = mcs.SetupTSOClient(suite.ctx, re, strings.Split(backendEndpoints, ","))
+		suite.client = mcs.SetupClientWithKeyspace(suite.ctx, re, strings.Split(backendEndpoints, ","))
 	}
 }