diff --git a/apps/sim/app/api/workflows/[id]/state/route.ts b/apps/sim/app/api/workflows/[id]/state/route.ts
index 9fad556412..1bb1b2d592 100644
--- a/apps/sim/app/api/workflows/[id]/state/route.ts
+++ b/apps/sim/app/api/workflows/[id]/state/route.ts
@@ -5,9 +5,9 @@ import { getSession } from '@/lib/auth'
 import { createLogger } from '@/lib/logs/console/logger'
 import { getUserEntityPermissions } from '@/lib/permissions/utils'
 import { saveWorkflowToNormalizedTables } from '@/lib/workflows/db-helpers'
+import { sanitizeAgentToolsInBlocks } from '@/lib/workflows/validation'
 import { db } from '@/db'
 import { workflow } from '@/db/schema'
-import { sanitizeAgentToolsInBlocks } from '@/lib/workflows/validation'
 
 const logger = createLogger('WorkflowStateAPI')
 
@@ -229,10 +229,7 @@ export async function PUT(request: NextRequest, { params }: { params: Promise<{
     const elapsed = Date.now() - startTime
     logger.info(`[${requestId}] Successfully saved workflow ${workflowId} state in ${elapsed}ms`)
 
-    return NextResponse.json(
-      { success: true, warnings },
-      { status: 200 }
-    )
+    return NextResponse.json({ success: true, warnings }, { status: 200 })
   } catch (error: any) {
     const elapsed = Date.now() - startTime
     logger.error(
@@ -241,7 +238,10 @@ export async function PUT(request: NextRequest, { params }: { params: Promise<{
     )
 
     if (error instanceof z.ZodError) {
-      return NextResponse.json({ error: 'Invalid request body', details: error.errors }, { status: 400 })
+      return NextResponse.json(
+        { error: 'Invalid request body', details: error.errors },
+        { status: 400 }
+      )
     }
 
     return NextResponse.json({ error: 'Internal server error' }, { status: 500 })
diff --git a/apps/sim/app/api/workflows/[id]/yaml/route.ts b/apps/sim/app/api/workflows/[id]/yaml/route.ts
index 980538fc8b..96f17d89b3 100644
--- a/apps/sim/app/api/workflows/[id]/yaml/route.ts
+++ b/apps/sim/app/api/workflows/[id]/yaml/route.ts
@@ -4,16 +4,19 @@ import { type NextRequest, NextResponse } from 'next/server'
 import { z } from 'zod'
 import { env } from '@/lib/env'
 import { createLogger } from '@/lib/logs/console/logger'
-import { db } from '@/db'
-import { workflow as workflowTable, workflowCheckpoints, customTools } from '@/db/schema'
+import { simAgentClient } from '@/lib/sim-agent'
+import {
+  loadWorkflowFromNormalizedTables,
+  saveWorkflowToNormalizedTables,
+} from '@/lib/workflows/db-helpers'
+import { sanitizeAgentToolsInBlocks } from '@/lib/workflows/validation'
+import { getUserId } from '@/app/api/auth/oauth/utils'
 import { getAllBlocks, getBlock } from '@/blocks'
 import type { BlockConfig } from '@/blocks/types'
-import { generateLoopBlocks, generateParallelBlocks } from '@/stores/workflows/workflow/utils'
 import { resolveOutputType } from '@/blocks/utils'
-import { getUserId } from '@/app/api/auth/oauth/utils'
-import { simAgentClient } from '@/lib/sim-agent'
-import { sanitizeAgentToolsInBlocks } from '@/lib/workflows/validation'
-import { loadWorkflowFromNormalizedTables, saveWorkflowToNormalizedTables } from '@/lib/workflows/db-helpers'
+import { db } from '@/db'
+import { customTools, workflowCheckpoints, workflow as workflowTable } from '@/db/schema'
+import { generateLoopBlocks, generateParallelBlocks } from '@/stores/workflows/workflow/utils'
 
 const logger = createLogger('YamlWorkflowAPI')
 
@@ -35,9 +38,7 @@ function updateBlockReferences(
     // Replace references in string values
     for (const [oldId, newId] of blockIdMapping.entries()) {
       if (value.includes(oldId)) {
-        value = value
-          .replaceAll(`<${oldId}.`, `<${newId}.`)
-          .replaceAll(`%${oldId}.`, `%${newId}.`)
+        value = value.replaceAll(`<${oldId}.`, `<${newId}.`).replaceAll(`%${oldId}.`, `%${newId}.`)
       }
     }
     return value
@@ -171,7 +172,11 @@ async function upsertCustomToolsFromBlocks(
           tool.schema.function.name &&
           typeof tool.code === 'string'
         ) {
-          collected.push({ title: tool.title || tool.schema.function.name, schema: tool.schema, code: tool.code })
+          collected.push({
+            title: tool.title || tool.schema.function.name,
+            schema: tool.schema,
+            code: tool.code,
+          })
         }
       }
     }
@@ -186,10 +191,7 @@ async function upsertCustomToolsFromBlocks(
     }
 
     // Load existing user's tools
-    const existing = await db
-      .select()
-      .from(customTools)
-      .where(eq(customTools.userId, userId))
+    const existing = await db.select().from(customTools).where(eq(customTools.userId, userId))
 
     const existingByName = new Map<string, (typeof existing)[number]>()
     for (const row of existing) {
@@ -325,8 +327,7 @@ export async function PUT(request: NextRequest, { params }: { params: Promise<{
 
     const conversionResult = await conversionResponse.json()
 
-    const workflowState =
-      conversionResult.workflowState || (conversionResult.diff && conversionResult.diff.proposedState)
+    const workflowState = conversionResult.workflowState || conversionResult.diff?.proposedState
 
     if (!conversionResult.success || !workflowState) {
       logger.error(`[${requestId}] YAML conversion failed`, {
@@ -639,9 +640,7 @@ export async function PUT(request: NextRequest, { params }: { params: Promise<{
       newWorkflowState.blocks
     )
     if (sanitationWarnings.length > 0) {
-      logger.warn(
-        `[${requestId}] Tool sanitation produced ${sanitationWarnings.length} warning(s)`
-      )
+      logger.warn(`[${requestId}] Tool sanitation produced ${sanitationWarnings.length} warning(s)`)
     }
     newWorkflowState.blocks = sanitizedBlocks
 
diff --git a/apps/sim/lib/workflows/db-helpers.ts b/apps/sim/lib/workflows/db-helpers.ts
index 2e715a4bf7..18e8af6b5b 100644
--- a/apps/sim/lib/workflows/db-helpers.ts
+++ b/apps/sim/lib/workflows/db-helpers.ts
@@ -1,10 +1,10 @@
 import { eq } from 'drizzle-orm'
 import { createLogger } from '@/lib/logs/console/logger'
+import { sanitizeAgentToolsInBlocks } from '@/lib/workflows/validation'
 import { db } from '@/db'
 import { workflow, workflowBlocks, workflowEdges, workflowSubflows } from '@/db/schema'
 import type { WorkflowState } from '@/stores/workflows/workflow/types'
 import { SUBFLOW_TYPES } from '@/stores/workflows/workflow/types'
-import { sanitizeAgentToolsInBlocks } from '@/lib/workflows/validation'
 
 const logger = createLogger('WorkflowDBHelpers')
 
diff --git a/apps/sim/lib/workflows/validation.ts b/apps/sim/lib/workflows/validation.ts
index 6c17ffebbb..e09abf3aeb 100644
--- a/apps/sim/lib/workflows/validation.ts
+++ b/apps/sim/lib/workflows/validation.ts
@@ -24,9 +24,10 @@ function isValidCustomToolSchema(tool: any): boolean {
   }
 }
 
-export function sanitizeAgentToolsInBlocks(
+export function sanitizeAgentToolsInBlocks(blocks: Record<string, any>): {
   blocks: Record<string, any>
-): { blocks: Record<string, any>; warnings: string[] } {
+  warnings: string[]
+} {
   const warnings: string[] = []
 
   // Shallow clone to avoid mutating callers
@@ -46,7 +47,9 @@ export function sanitizeAgentToolsInBlocks(
         try {
           value = JSON.parse(value)
         } catch (_e) {
-          warnings.push(`Block ${block.name || blockId}: invalid tools JSON; resetting tools to empty array`)
+          warnings.push(
+            `Block ${block.name || blockId}: invalid tools JSON; resetting tools to empty array`
+          )
           value = []
         }
       }
@@ -103,4 +106,4 @@ export function sanitizeAgentToolsInBlocks(
   }
 
   return { blocks: sanitizedBlocks, warnings }
-} 
\ No newline at end of file
+}