diff --git a/go.mod b/go.mod
index f4d5da2..460beba 100644
--- a/go.mod
+++ b/go.mod
@@ -5,7 +5,7 @@ go 1.13
 require (
 	github.com/RoaringBitmap/roaring v0.5.0
 	github.com/alldroll/cdb v1.0.2
-	github.com/alldroll/go-datastructures v0.0.0-20190322060030-1d3a19ff3b29
+	github.com/alldroll/rbtree v0.0.0-20201026153457-c76906afcaa0
 	github.com/edsrzf/mmap-go v0.0.0-20190108065903-904c4ced31cd
 	github.com/gorilla/handlers v1.4.0
 	github.com/gorilla/mux v1.7.1
diff --git a/go.sum b/go.sum
index 81b58cd..2f979df 100644
--- a/go.sum
+++ b/go.sum
@@ -2,8 +2,8 @@ github.com/RoaringBitmap/roaring v0.5.0 h1:0psZZWU0J2AUl29BAylpHAsuBEEhCEfTKl2v5
 github.com/RoaringBitmap/roaring v0.5.0/go.mod h1:D0gp8kJQgE1A4LQ5wFLggQEyvDi06Mq5mKs52e1TwOo=
 github.com/alldroll/cdb v1.0.2 h1:pSB3BphsF0m2DqOZm+IFyNm38nz1R8kCg3DPCusPLQE=
 github.com/alldroll/cdb v1.0.2/go.mod h1:PK3VAN9pconusJqa4kzOupYg9QxOnmgU8AcBWhuZZdo=
-github.com/alldroll/go-datastructures v0.0.0-20190322060030-1d3a19ff3b29 h1:gKZgtn2ud0FxyG0lFqrp8hRnvABoqkDTNgebDlbmtgM=
-github.com/alldroll/go-datastructures v0.0.0-20190322060030-1d3a19ff3b29/go.mod h1:3IP5cUVnXIyZIsUbKXztqoz1brdOXngjFyYtmwqz9MY=
+github.com/alldroll/rbtree v0.0.0-20201026153457-c76906afcaa0 h1:IRs8Y64CCc/GWRo0a4+NiWyFjF6TfRO5iZKTXyCM5B0=
+github.com/alldroll/rbtree v0.0.0-20201026153457-c76906afcaa0/go.mod h1:iBiS1ITTL31hmJ3cDRrtawPChwBaFPQDTQOpGhSF418=
 github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/edsrzf/mmap-go v0.0.0-20190108065903-904c4ced31cd h1:v8VTjPes659sdlQ3O2AbICsk2XjORhYc76QLCFSTEgA=
diff --git a/pkg/lm/binary.go b/pkg/lm/binary.go
index 2db1c7d..5448602 100644
--- a/pkg/lm/binary.go
+++ b/pkg/lm/binary.go
@@ -7,7 +7,7 @@ import (
 	"strconv"
 	"strings"
 
-	"github.com/alldroll/go-datastructures/rbtree"
+	"github.com/alldroll/rbtree"
 	"github.com/suggest-go/suggest/pkg/dictionary"
 	"github.com/suggest-go/suggest/pkg/mph"
 	"github.com/suggest-go/suggest/pkg/store"
diff --git a/pkg/lm/ngram_vector_builder.go b/pkg/lm/ngram_vector_builder.go
index 3110c75..94f2592 100644
--- a/pkg/lm/ngram_vector_builder.go
+++ b/pkg/lm/ngram_vector_builder.go
@@ -5,7 +5,7 @@ import (
 	"fmt"
 	"log"
 
-	"github.com/alldroll/go-datastructures/rbtree"
+	"github.com/alldroll/rbtree"
 	"github.com/suggest-go/suggest/pkg/utils"
 )