diff --git a/src/services/checkpoints/__tests__/ShadowCheckpointService.spec.ts b/src/services/checkpoints/__tests__/ShadowCheckpointService.spec.ts
index a403c4e851..cba8eee8ba 100644
--- a/src/services/checkpoints/__tests__/ShadowCheckpointService.spec.ts
+++ b/src/services/checkpoints/__tests__/ShadowCheckpointService.spec.ts
@@ -12,8 +12,6 @@ import * as fileSearch from "../../../services/search/file-search"
 
 import { RepoPerTaskCheckpointService } from "../RepoPerTaskCheckpointService"
 
-vitest.setConfig({ testTimeout: 20_000 })
-
 const tmpDir = path.join(os.tmpdir(), "CheckpointService")
 
 const initWorkspaceRepo = async ({
diff --git a/src/vitest.config.ts b/src/vitest.config.ts
index e20e40c655..4e576f3c96 100644
--- a/src/vitest.config.ts
+++ b/src/vitest.config.ts
@@ -8,6 +8,8 @@ export default defineConfig({
 		watch: false,
 		reporters: ["dot"],
 		silent: true,
+		testTimeout: 20_000,
+		hookTimeout: 20_000,
 	},
 	resolve: {
 		alias: {