diff --git a/cache.go b/cache.go
index 86ef5813..d29c93d0 100644
--- a/cache.go
+++ b/cache.go
@@ -28,7 +28,7 @@ import (
 	"time"
 	"unsafe"
 
-	"github.com/dgraph-io/ristretto/z"
+	"github.com/dgraph-io/ristretto/v2/z"
 )
 
 var (
diff --git a/cache_test.go b/cache_test.go
index 2a4eb422..263c1193 100644
--- a/cache_test.go
+++ b/cache_test.go
@@ -10,7 +10,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/dgraph-io/ristretto/z"
+	"github.com/dgraph-io/ristretto/v2/z"
 	"github.com/stretchr/testify/require"
 )
 
diff --git a/contrib/demo/node.go b/contrib/demo/node.go
index 3138437b..400130b1 100644
--- a/contrib/demo/node.go
+++ b/contrib/demo/node.go
@@ -4,8 +4,9 @@ import (
 	"fmt"
 	"runtime"
 
-	"github.com/dgraph-io/ristretto/z"
 	"github.com/dustin/go-humanize"
+
+	"github.com/dgraph-io/ristretto/v2/z"
 )
 
 type node struct {
diff --git a/contrib/demo/node_allocator.go b/contrib/demo/node_allocator.go
index 2db63de1..f6ab8387 100644
--- a/contrib/demo/node_allocator.go
+++ b/contrib/demo/node_allocator.go
@@ -6,7 +6,7 @@ package main
 import (
 	"unsafe"
 
-	"github.com/dgraph-io/ristretto/z"
+	"github.com/dgraph-io/ristretto/v2/z"
 )
 
 // Defined in node.go.
diff --git a/contrib/demo/node_jemalloc.go b/contrib/demo/node_jemalloc.go
index 34752dd0..1f253b4b 100644
--- a/contrib/demo/node_jemalloc.go
+++ b/contrib/demo/node_jemalloc.go
@@ -6,7 +6,7 @@ package main
 import (
 	"unsafe"
 
-	"github.com/dgraph-io/ristretto/z"
+	"github.com/dgraph-io/ristretto/v2/z"
 )
 
 func newNode(val int) *node {
diff --git a/contrib/memtest/main.go b/contrib/memtest/main.go
index 565c046e..2e51a4f3 100644
--- a/contrib/memtest/main.go
+++ b/contrib/memtest/main.go
@@ -34,7 +34,7 @@ import (
 
 	"github.com/dustin/go-humanize"
 
-	"github.com/dgraph-io/ristretto/z"
+	"github.com/dgraph-io/ristretto/v2/z"
 )
 
 type S struct {
diff --git a/contrib/memtest/withjemalloc.go b/contrib/memtest/withjemalloc.go
index ed1f1d98..5c379c8f 100644
--- a/contrib/memtest/withjemalloc.go
+++ b/contrib/memtest/withjemalloc.go
@@ -6,7 +6,7 @@ package main
 import (
 	"log"
 
-	"github.com/dgraph-io/ristretto/z"
+	"github.com/dgraph-io/ristretto/v2/z"
 )
 
 func Calloc(size int) []byte { return z.Calloc(size, "memtest") }
diff --git a/go.mod b/go.mod
index 43602072..6b33a1d2 100644
--- a/go.mod
+++ b/go.mod
@@ -1,4 +1,4 @@
-module github.com/dgraph-io/ristretto
+module github.com/dgraph-io/ristretto/v2
 
 go 1.23
 
@@ -16,10 +16,3 @@ require (
 	github.com/pmezard/go-difflib v1.0.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
-
-// we retract v1.0.0 because v0.2.0 is not backwards compatible with v1.0.0.
-// The users should upgrade to v2.0.0 instead once released.
-retract v1.0.0
-
-// need to retract the next release as well
-retract v1.0.1
diff --git a/policy.go b/policy.go
index 08963b1e..d4e340ad 100644
--- a/policy.go
+++ b/policy.go
@@ -21,7 +21,7 @@ import (
 	"sync"
 	"sync/atomic"
 
-	"github.com/dgraph-io/ristretto/z"
+	"github.com/dgraph-io/ristretto/v2/z"
 )
 
 const (
diff --git a/store_test.go b/store_test.go
index c912e6c3..2cc99004 100644
--- a/store_test.go
+++ b/store_test.go
@@ -4,7 +4,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/dgraph-io/ristretto/z"
+	"github.com/dgraph-io/ristretto/v2/z"
 	"github.com/stretchr/testify/require"
 )
 
diff --git a/stress_test.go b/stress_test.go
index cf0ec07d..641648c0 100644
--- a/stress_test.go
+++ b/stress_test.go
@@ -9,7 +9,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/dgraph-io/ristretto/sim"
+	"github.com/dgraph-io/ristretto/v2/sim"
 	"github.com/stretchr/testify/require"
 )
 
diff --git a/z/btree.go b/z/btree.go
index 33c3046b..a8a83816 100644
--- a/z/btree.go
+++ b/z/btree.go
@@ -24,7 +24,7 @@ import (
 	"strings"
 	"unsafe"
 
-	"github.com/dgraph-io/ristretto/z/simd"
+	"github.com/dgraph-io/ristretto/v2/z/simd"
 )
 
 var (
diff --git a/z/btree_test.go b/z/btree_test.go
index ea7410a6..a6100d73 100644
--- a/z/btree_test.go
+++ b/z/btree_test.go
@@ -26,9 +26,10 @@ import (
 	"testing"
 	"time"
 
-	"github.com/dgraph-io/ristretto/z/simd"
 	"github.com/dustin/go-humanize"
 	"github.com/stretchr/testify/require"
+
+	"github.com/dgraph-io/ristretto/v2/z/simd"
 )
 
 var tmp int