diff --git a/integration_tests/raw/api_test.go b/integration_tests/raw/api_test.go
index 4f86387cd..a95175beb 100644
--- a/integration_tests/raw/api_test.go
+++ b/integration_tests/raw/api_test.go
@@ -91,7 +91,7 @@ func (s *apiTestSuite) SetupTest() {
 	s.Nil(err)
 	s.apiVersion = s.getApiVersion(pdClient)
 
-	s.pdClient = s.wrapPDClient(pdClient, addrs)
+	s.pdClient = s.wrapPDClient(pdClient)
 
 	client := s.newRawKVClient(pdClient, addrs)
 	s.client = client