diff --git a/benchmarks/benchmarks_test.go b/benchmarks/benchmarks_test.go
index 47a5c3c..2be1771 100644
--- a/benchmarks/benchmarks_test.go
+++ b/benchmarks/benchmarks_test.go
@@ -10,6 +10,7 @@ import (
 	"github.com/jmoiron/sqlx"
 	_ "github.com/lib/pq"
 	"github.com/vingarcia/ksql"
+	"github.com/vingarcia/ksql/kpgx"
 )
 
 var UsersTable = ksql.NewTable("users")
@@ -52,7 +53,7 @@ func BenchmarkInsert(b *testing.B) {
 		})
 	})
 
-	pgxDB, err := ksql.NewWithPGX(ctx, connStr, ksql.Config{
+	pgxDB, err := kpgx.New(ctx, connStr, ksql.Config{
 		MaxOpenConns: 1,
 	})
 	if err != nil {
diff --git a/benchmarks/go.mod b/benchmarks/go.mod
index 4e492cf..7b5a222 100644
--- a/benchmarks/go.mod
+++ b/benchmarks/go.mod
@@ -7,3 +7,5 @@ require (
 	github.com/lib/pq v1.10.4
 	github.com/vingarcia/ksql v1.1.0
 )
+
+replace github.com/vingarcia/ksql => ../