diff --git a/sources/app/social/friendAdd.ts b/sources/app/social/friendAdd.ts
index f4467ec..5746ea5 100644
--- a/sources/app/social/friendAdd.ts
+++ b/sources/app/social/friendAdd.ts
@@ -1,6 +1,6 @@
 import { Context } from "@/context";
 import { buildUserProfile, UserProfile } from "./type";
-import { db } from "@/storage/db";
+import { inTx } from "@/storage/inTx";
 import { RelationshipStatus } from "@prisma/client";
 import { relationshipSet } from "./relationshipSet";
 import { relationshipGet } from "./relationshipGet";
@@ -20,7 +20,7 @@ export async function friendAdd(ctx: Context, uid: string): Promise<UserProfile
     }
 
     // Update relationship status
-    return await db.$transaction(async (tx) => {
+    return await inTx(async (tx) => {
 
         // Read current user objects
         const currentUser = await tx.account.findUnique({
diff --git a/sources/app/social/friendRemove.ts b/sources/app/social/friendRemove.ts
index afb819a..0063d8c 100644
--- a/sources/app/social/friendRemove.ts
+++ b/sources/app/social/friendRemove.ts
@@ -1,12 +1,12 @@
 import { Context } from "@/context";
 import { buildUserProfile, UserProfile } from "./type";
-import { db } from "@/storage/db";
+import { inTx } from "@/storage/inTx";
 import { RelationshipStatus } from "@prisma/client";
 import { relationshipSet } from "./relationshipSet";
 import { relationshipGet } from "./relationshipGet";
 
 export async function friendRemove(ctx: Context, uid: string): Promise<UserProfile | null> {
-    return await db.$transaction(async (tx) => {
+    return await inTx(async (tx) => {
 
         // Read current user objects
         const currentUser = await tx.account.findUnique({