diff --git a/database/middleware.go b/database/middleware.go
index 364a9554c7da8b676ce86e885370fd2a86f00ae5..adbf56bf0852e5984251d349349ac5e12242f1ab 100644
--- a/database/middleware.go
+++ b/database/middleware.go
@@ -44,12 +44,8 @@ 	return raw
 }
 
 func WithTx(ctx context.Context, opts *sql.TxOptions, fn func(tx *sql.Tx) error) error {
-	conn, err := ForContext(ctx)
-	if err != nil {
-		return err
-	}
-	defer conn.Close()
-	tx, err := conn.BeginTx(ctx, opts)
+	db := DBForContext(ctx)
+	tx, err := db.BeginTx(ctx, opts)
 	if err != nil {
 		return err
 	}