diff --git a/packages/agent-runtime/src/__tests__/read-docs-tool.test.ts b/packages/agent-runtime/src/__tests__/read-docs-tool.test.ts
index 77898e0bee..fc3bbf5ac3 100644
--- a/packages/agent-runtime/src/__tests__/read-docs-tool.test.ts
+++ b/packages/agent-runtime/src/__tests__/read-docs-tool.test.ts
@@ -14,7 +14,8 @@ import {
 } from 'bun:test'
 
 import { createToolCallChunk, mockFileContext } from './test-utils'
-import researcherAgent from '../../../../agents-graveyard/researcher/researcher'
+import researcherDocsAgent from '../../../../agents/researcher/researcher-docs'
+const researcherAgent = { ...researcherDocsAgent, id: 'researcher' }
 import * as webApi from '../llm-api/codebuff-web-api'
 import { runAgentStep } from '../run-agent-step'
 import { assembleLocalAgentTemplates } from '../templates/agent-registry'
diff --git a/packages/agent-runtime/src/__tests__/web-search-tool.test.ts b/packages/agent-runtime/src/__tests__/web-search-tool.test.ts
index 69145b6561..0f23d7eaeb 100644
--- a/packages/agent-runtime/src/__tests__/web-search-tool.test.ts
+++ b/packages/agent-runtime/src/__tests__/web-search-tool.test.ts
@@ -14,7 +14,8 @@ import {
 } from 'bun:test'
 
 import { createToolCallChunk, mockFileContext } from './test-utils'
-import researcherAgent from '../../../../agents-graveyard/researcher/researcher'
+import researcherWebAgent from '../../../../agents/researcher/researcher-web'
+const researcherAgent = { ...researcherWebAgent, id: 'researcher' }
 import * as webApi from '../llm-api/codebuff-web-api'
 import { runAgentStep } from '../run-agent-step'
 import { assembleLocalAgentTemplates } from '../templates/agent-registry'