diff --git a/internal/querycoord/impl_test.go b/internal/querycoord/impl_test.go
index 901938a935..18b2739685 100644
--- a/internal/querycoord/impl_test.go
+++ b/internal/querycoord/impl_test.go
@@ -910,11 +910,18 @@ func Test_RepeatedLoadDifferentPartitions(t *testing.T) {
 	status, err := queryCoord.LoadPartitions(ctx, loadPartitionReq)
 	assert.Equal(t, commonpb.ErrorCode_Success, status.ErrorCode)
 	assert.Nil(t, err)
-	waitLoadPartitionDone(ctx, queryCoord, defaultCollectionID, []UniqueID{defaultPartitionID})
+	assert.Nil(t, waitLoadPartitionDone(ctx, queryCoord, defaultCollectionID, []UniqueID{defaultPartitionID}))
 
 	// second load defaultPartitionID+1
-	loadPartitionReq.PartitionIDs = []UniqueID{defaultPartitionID + 1}
-	status, err = queryCoord.LoadPartitions(ctx, loadPartitionReq)
+	failLoadRequest := &querypb.LoadPartitionsRequest{
+		Base: &commonpb.MsgBase{
+			MsgType: commonpb.MsgType_LoadPartitions,
+		},
+		CollectionID: defaultCollectionID,
+		PartitionIDs: []UniqueID{defaultPartitionID + 1},
+		Schema:       genDefaultCollectionSchema(false),
+	}
+	status, err = queryCoord.LoadPartitions(ctx, failLoadRequest)
 	assert.Equal(t, commonpb.ErrorCode_UnexpectedError, status.ErrorCode)
 	assert.Nil(t, err)