diff --git a/db_test.go b/db_test.go
index 0e63687..fe84eae 100644
--- a/db_test.go
+++ b/db_test.go
@@ -1,4 +1,4 @@
-package pgtxdb
+package pgtxdb_test
 
 import (
 	"database/sql"
@@ -8,10 +8,11 @@ import (
 	"testing"
 
 	_ "github.com/jackc/pgx/v5/stdlib" // pgx
+	"github.com/kanmu/pgtxdb"
 )
 
 func init() {
-	Register("pgtxdb", "pgx", "postgres://pgtxdbtest@localhost:5432/pgtxdbtest?sslmode=disable")
+	pgtxdb.Register("pgtxdb", "pgx", "postgres://pgtxdbtest@localhost:5432/pgtxdbtest?sslmode=disable")
 }
 
 func TestShouldRunWithinTransaction(t *testing.T) {
diff --git a/main_test.go b/main_test.go
index ec8a839..b1d016d 100644
--- a/main_test.go
+++ b/main_test.go
@@ -1,4 +1,4 @@
-package pgtxdb
+package pgtxdb_test
 
 import (
 	"database/sql"