diff --git a/internal/querycoordv2/balance/streaming_query_node_channel_helper_test.go b/internal/querycoordv2/balance/streaming_query_node_channel_helper_test.go
index 0cbf73359cb8a..492b64ddc6565 100644
--- a/internal/querycoordv2/balance/streaming_query_node_channel_helper_test.go
+++ b/internal/querycoordv2/balance/streaming_query_node_channel_helper_test.go
@@ -4,6 +4,9 @@ import (
 	"context"
 	"testing"
 
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+
 	"github.com/milvus-io/milvus/internal/coordinator/snmanager"
 	"github.com/milvus-io/milvus/internal/mocks/streamingcoord/server/mock_balancer"
 	"github.com/milvus-io/milvus/internal/proto/datapb"
@@ -11,8 +14,6 @@ import (
 	"github.com/milvus-io/milvus/internal/querycoordv2/session"
 	"github.com/milvus-io/milvus/pkg/streaming/util/types"
 	"github.com/milvus-io/milvus/pkg/util/typeutil"
-	"github.com/stretchr/testify/assert"
-	"github.com/stretchr/testify/mock"
 )
 
 func TestAssignChannelToWALLocatedFirst(t *testing.T) {