diff --git a/internal/proxynode/insert_channels.go b/internal/proxynode/insert_channels.go
index 90451614bdfd208157da1fe81409fc869d943b0c..deda7ef03c957858ebf3a92ad92c1d894d22aa3c 100644
--- a/internal/proxynode/insert_channels.go
+++ b/internal/proxynode/insert_channels.go
@@ -134,10 +134,12 @@ func (m *InsertChannelsMap) closeInsertMsgStream(collID UniqueID) error {
 	m.usageHistogram[loc]--
 	if m.usageHistogram[loc] <= 0 {
 		m.insertMsgStreams[loc].Close()
-		m.droppedBitMap[loc] = 1
-		delete(m.collectionID2InsertChannels, collID)
-		log.Print("close insert message stream ...")
 	}
+	log.Print("close insert message stream ...")
+
+	m.droppedBitMap[loc] = 1
+	delete(m.collectionID2InsertChannels, collID)
+
 	return nil
 }
 
diff --git a/internal/querynode/param_table.go b/internal/querynode/param_table.go
index a2254f1bdb710f62e83878171aca97c13860e050..d34a04e6ea3005a96cd7f9dcb034c7c4eada1701 100644
--- a/internal/querynode/param_table.go
+++ b/internal/querynode/param_table.go
@@ -4,8 +4,8 @@ import (
 	"log"
 	"os"
 	"strconv"
-	"strings"
 
+	"github.com/zilliztech/milvus-distributed/internal/util/funcutil"
 	"github.com/zilliztech/milvus-distributed/internal/util/paramtable"
 )
 
@@ -87,22 +87,13 @@ func (p *ParamTable) Init() {
 		}
 	}
 
-	queryNodeAddress := os.Getenv("QUERY_NODE_ADDRESS")
-	if queryNodeAddress == "" {
+	queryNodeIP := os.Getenv("QUERY_NODE_IP")
+	if queryNodeIP == "" {
 		p.QueryNodeIP = "localhost"
-		p.QueryNodePort = 20010
 	} else {
-		ipAndPort := strings.Split(queryNodeAddress, ":")
-		if len(ipAndPort) != 2 {
-			panic("illegal query node address")
-		}
-		p.QueryNodeIP = ipAndPort[0]
-		port, err := strconv.ParseInt(ipAndPort[1], 10, 64)
-		if err != nil {
-			panic(err)
-		}
-		p.QueryNodePort = port
+		p.QueryNodeIP = queryNodeIP
 	}
+	p.QueryNodePort = int64(funcutil.GetAvailablePort())
 
 	err = p.LoadYaml("advanced/common.yaml")
 	if err != nil {
diff --git a/tests/python/requirements.txt b/tests/python/requirements.txt
index a4136e23e77e33a092b21452c4cfad575d97d6b7..fcf08cf5d07356c9e2a31351229ead6dd82400a3 100644
--- a/tests/python/requirements.txt
+++ b/tests/python/requirements.txt
@@ -2,7 +2,7 @@ grpcio==1.26.0
 grpcio-tools==1.26.0
 numpy==1.18.1
 pytest-cov==2.8.1
-pymilvus-distributed==0.0.17
+pymilvus-distributed==0.0.19
 sklearn==0.0
 pytest==4.5.0
 pytest-timeout==1.3.3