diff --git a/go.mod b/go.mod
index 3a5e935..1eb0feb 100644
--- a/go.mod
+++ b/go.mod
@@ -11,8 +11,7 @@ require (
 	github.com/prometheus/client_golang v1.9.0 // indirect
 	github.com/satori/go.uuid v1.2.0
 	github.com/shiena/ansicolor v0.0.0-20200904210342-c7312218db18 // indirect
-	github.com/stretchr/testify v1.7.0 // indirect
-	github.com/vesoft-inc/nebula-go/v2 v2.5.0
+	github.com/vesoft-inc/nebula-go/v2 v2.5.1
 	github.com/vesoft-inc/nebula-importer v0.0.0-20210716031041-8882282482b2
 	golang.org/x/crypto v0.0.0-20201221181555-eec23a3978ad // indirect
 	golang.org/x/net v0.0.0-20201224014010-6772e930b67b // indirect
diff --git a/go.sum b/go.sum
index b0dbfc2..e9e45b2 100644
--- a/go.sum
+++ b/go.sum
@@ -319,8 +319,9 @@ github.com/tmc/grpc-websocket-proxy v0.0.0-20170815181823-89b8d40f7ca8/go.mod h1
 github.com/ugorji/go v0.0.0-20171122102828-84cb69a8af83/go.mod h1:hnLbHMwcvSihnDhEfx2/BzKp2xb0Y+ErdfYcrs9tkJQ=
 github.com/urfave/cli v1.20.0/go.mod h1:70zkFmudgCuE/ngEzBv17Jvp/497gISqfk5gWijbERA=
 github.com/urfave/cli v1.22.1/go.mod h1:Gos4lmkARVdJ6EkW0WaNv/tZAAMe9V7XWyB60NtXRu0=
-github.com/vesoft-inc/nebula-go/v2 v2.5.0 h1:A4wZoGqR1W1mQ9y/X+tV1wCMIsDkuhFMthTQKMeT8Yc=
-github.com/vesoft-inc/nebula-go/v2 v2.5.0/go.mod h1:fehDUs97/mpmxXi9WezhznX0Dg7hmQRUoOWgDZv9zG0=
+github.com/vesoft-inc/nebula-go/v2 v2.0.0-20210701060243-a0577f67f375/go.mod h1:B7nR6+nOSo0umq/HkCmUfyRtYrJVOsNiPS9u4djDbSc=
+github.com/vesoft-inc/nebula-go/v2 v2.5.1 h1:y4gBQzRqb3W4jeBbbAdtCPxVZ4tXKv9TJLHjNuWGhI0=
+github.com/vesoft-inc/nebula-go/v2 v2.5.1/go.mod h1:fehDUs97/mpmxXi9WezhznX0Dg7hmQRUoOWgDZv9zG0=
 github.com/vesoft-inc/nebula-importer v0.0.0-20210716031041-8882282482b2 h1:TJnK8RiJeFxzgo37iAC4roGeotieejr/EEryiaSSSo8=
 github.com/vesoft-inc/nebula-importer v0.0.0-20210716031041-8882282482b2/go.mod h1:kHxOwdaHWgpypNvGK/biKn4+mVWT28o51sGmZ1YdOjA=
 github.com/wendal/errors v0.0.0-20130201093226-f66c77a7882b/go.mod h1:Q12BUT7DqIlHRmgv3RskH+UCM/4eqVMgI0EMmlSpAXc=