diff --git a/tests/api-admin-devices-rename.test.ts b/tests/api-admin-devices-rename.test.ts
index e264c3a..8894e95 100644
--- a/tests/api-admin-devices-rename.test.ts
+++ b/tests/api-admin-devices-rename.test.ts
@@ -13,16 +13,17 @@ vi.mock("convex/browser", () => ({
   ConvexHttpClient: vi.fn(() => ({ mutation: mutationMock })),
 }))
 
-vi.mock("@/convex/_generated/api", async () => {
-  const actual = await import("@/convex/_generated/api")
-  return {
-    ...actual,
-    api: {
-      ...actual.api,
-      users: { ensureUser: "users:ensureUser" },
+vi.mock("@/convex/_generated/api", () => ({
+  api: {
+    users: { ensureUser: "users:ensureUser" },
+    devices: {
+      heartbeat: "devices:heartbeat",
+      upsertInventory: "devices:upsertInventory",
+      findByAuthEmail: "devices:findByAuthEmail",
+      remove: "devices:remove",
     },
-  }
-})
+  },
+}))
 
 describe("POST /api/admin/devices/rename", () => {
   beforeEach(() => {