diff --git a/internal/querycoordv2/server_test.go b/internal/querycoordv2/server_test.go
index 1c96f1282f..eaf5cf8932 100644
--- a/internal/querycoordv2/server_test.go
+++ b/internal/querycoordv2/server_test.go
@@ -241,7 +241,7 @@ func (suite *ServerSuite) TestEnableActiveStandby() {
 				),
 				CollectionID: collection,
 			}
-			mockRootCoord.EXPECT().ShowPartitions(mock.Anything, req).Return(&milvuspb.ShowPartitionsResponse{
+			mockRootCoord.EXPECT().ShowPartitionsInternal(mock.Anything, req).Return(&milvuspb.ShowPartitionsResponse{
 				Status:       successStatus,
 				PartitionIDs: suite.partitions[collection],
 			}, nil).Maybe()