diff --git a/tests/client/profiles-store.test.ts b/tests/client/profiles-store.test.ts
index 72da72b..283bcfe 100644
--- a/tests/client/profiles-store.test.ts
+++ b/tests/client/profiles-store.test.ts
@@ -52,16 +52,16 @@ describe('Profiles Store', () => {
   })
 
   it('createProfile calls API and refreshes list', async () => {
-    mockProfilesApi.createProfile.mockResolvedValue(true)
+    mockProfilesApi.createProfile.mockResolvedValue({ success: true })
     mockProfilesApi.fetchProfiles.mockResolvedValue([
       { name: 'default', active: true, model: 'gpt-4', gateway: 'running', alias: '' },
       { name: 'new-profile', active: false, model: 'gpt-4', gateway: 'stopped', alias: '' },
     ])
 
     const store = useProfilesStore()
-    const ok = await store.createProfile('new-profile', false)
+    const result = await store.createProfile('new-profile', false)
 
-    expect(ok).toBe(true)
+    expect(result.success).toBe(true)
     expect(mockProfilesApi.createProfile).toHaveBeenCalledWith('new-profile', false)
     expect(store.profiles).toHaveLength(2)
   })
diff --git a/tests/server/sessions-routes.test.ts b/tests/server/sessions-routes.test.ts
index 2270702..48e3290 100644
--- a/tests/server/sessions-routes.test.ts
+++ b/tests/server/sessions-routes.test.ts
@@ -8,6 +8,8 @@ const searchMock = vi.fn(async (ctx: any) => { ctx.body = { results: [{ id: 'sea
 const getMock = vi.fn(async (ctx: any) => { ctx.body = { session: { id: ctx.params.id } } })
 const removeMock = vi.fn(async (ctx: any) => { ctx.body = { ok: true } })
 const renameMock = vi.fn(async (ctx: any) => { ctx.body = { ok: true } })
+const setWorkspaceMock = vi.fn(async (ctx: any) => { ctx.body = { ok: true } })
+const listWorkspaceFoldersMock = vi.fn(async (ctx: any) => { ctx.body = { folders: [] } })
 const usageBatchMock = vi.fn(async (ctx: any) => { ctx.body = {} })
 const usageSingleMock = vi.fn(async (ctx: any) => { ctx.body = { input_tokens: 0, output_tokens: 0 } })
 const usageStatsMock = vi.fn(async (ctx: any) => { ctx.body = { total_input_tokens: 0, total_output_tokens: 0 } })
@@ -22,6 +24,8 @@ vi.mock('../../packages/server/src/controllers/hermes/sessions', () => ({
   get: getMock,
   remove: removeMock,
   rename: renameMock,
+  setWorkspace: setWorkspaceMock,
+  listWorkspaceFolders: listWorkspaceFoldersMock,
   usageBatch: usageBatchMock,
   usageSingle: usageSingleMock,
   usageStats: usageStatsMock,
diff --git a/tests/server/usage-store.test.ts b/tests/server/usage-store.test.ts
index 15454ce..f4afaa1 100644
--- a/tests/server/usage-store.test.ts
+++ b/tests/server/usage-store.test.ts
@@ -20,7 +20,6 @@ vi.mock('../../packages/server/src/db/index', () => ({
 }))
 
 import {
-  initUsageStore,
   updateUsage,
   getUsage,
   getUsageBatch,
@@ -32,12 +31,6 @@ describe('Usage Store (JSON fallback)', () => {
     vi.clearAllMocks()
   })
 
-  it('initUsageStore calls ensureTable when SQLite is available', () => {
-    // In our mock, isSqliteAvailable returns false, so ensureTable should NOT be called
-    initUsageStore()
-    expect(mockEnsureTable).not.toHaveBeenCalled()
-  })
-
   it('updateUsage writes via jsonSet', () => {
     updateUsage('session-1', { inputTokens: 100, outputTokens: 50 })
     expect(mockJsonSet).toHaveBeenCalledWith(