diff --git a/src/main/apiServer/routes/agents/handlers/messages.ts b/src/main/apiServer/routes/agents/handlers/messages.ts
index 2cc8668e07..e18fadc0e0 100644
--- a/src/main/apiServer/routes/agents/handlers/messages.ts
+++ b/src/main/apiServer/routes/agents/handlers/messages.ts
@@ -1,10 +1,9 @@
 import { loggerService } from '@logger'
+import { MESSAGE_STREAM_TIMEOUT_MS } from '@main/apiServer/config/timeouts'
+import { createStreamAbortController, STREAM_TIMEOUT_REASON } from '@main/apiServer/utils/createStreamAbortController'
+import { agentService, sessionMessageService, sessionService } from '@main/services/agents'
 import { Request, Response } from 'express'
 
-import { agentService, sessionMessageService, sessionService } from '../../../../services/agents'
-import { MESSAGE_STREAM_TIMEOUT_MS } from '../../../config/timeouts'
-import { createStreamAbortController, STREAM_TIMEOUT_REASON } from '../../../utils/createStreamAbortController'
-
 const logger = loggerService.withContext('ApiServerMessagesHandlers')
 
 // Helper function to verify agent and session exist and belong together