diff --git a/apps/server/src/api/v2/deploy/POST.test.ts b/apps/server/src/api/v2/deploy/POST.test.ts
index 20af1031c..a5a2f0bec 100644
--- a/apps/server/src/api/v2/deploy/POST.test.ts
+++ b/apps/server/src/api/v2/deploy/POST.test.ts
@@ -10,6 +10,9 @@ vi.mock('@buster/database/queries', () => ({
   getDataSourceByName: vi.fn(),
   upsertDataset: vi.fn(),
   upsertDoc: vi.fn(),
+  deleteLogsWriteBackConfig: vi.fn(),
+  getDataSourceCredentials: vi.fn(),
+  upsertLogsWriteBackConfig: vi.fn(),
 }));
 
 vi.mock('@buster/database/connection', () => ({