diff --git a/server/routers/target/createTarget.ts b/server/routers/target/createTarget.ts
index 7a3acd55..f58c236e 100644
--- a/server/routers/target/createTarget.ts
+++ b/server/routers/target/createTarget.ts
@@ -161,7 +161,7 @@ export async function createTarget(
                 );
             }
 
-            const { internalPort, targetIps } = await pickPort(site.siteId!);
+            const { internalPort, targetIps } = await pickPort(site.siteId!, db);
 
             if (!internalPort) {
                 return next(
diff --git a/server/routers/target/helpers.ts b/server/routers/target/helpers.ts
index 4935d28a..13b2ee46 100644
--- a/server/routers/target/helpers.ts
+++ b/server/routers/target/helpers.ts
@@ -1,10 +1,10 @@
-import { db } from "@server/db";
+import { db, Transaction } from "@server/db";
 import { resources, targets } from "@server/db";
 import { eq } from "drizzle-orm";
 
 const currentBannedPorts: number[] = [];
 
-export async function pickPort(siteId: number): Promise<{
+export async function pickPort(siteId: number, trx: Transaction | typeof db): Promise<{
     internalPort: number;
     targetIps: string[];
 }> {
@@ -12,7 +12,7 @@ export async function pickPort(siteId: number): Promise<{
     const targetIps: string[] = [];
     const targetInternalPorts: number[] = [];
 
-    const targetsRes = await db
+    const targetsRes = await trx
         .select()
         .from(targets)
         .where(eq(targets.siteId, siteId));
diff --git a/server/routers/target/updateTarget.ts b/server/routers/target/updateTarget.ts
index 67d9a8df..47300619 100644
--- a/server/routers/target/updateTarget.ts
+++ b/server/routers/target/updateTarget.ts
@@ -153,7 +153,7 @@ export async function updateTarget(
             );
         }
 
-        const { internalPort, targetIps } = await pickPort(site.siteId!);
+        const { internalPort, targetIps } = await pickPort(site.siteId!, db);
 
         if (!internalPort) {
             return next(