diff --git a/extension/src/service-worker/router/__tests__/router.test.ts b/extension/src/service-worker/router/__tests__/router.test.ts
index 117610f..0c7dbbe 100644
--- a/extension/src/service-worker/router/__tests__/router.test.ts
+++ b/extension/src/service-worker/router/__tests__/router.test.ts
@@ -378,18 +378,18 @@ describe('setup tab exception scope', () => {
 
 // --- register_this_device: wasm returns a JS object, not a JSON string ---
 //
-// The #[wasm_bindgen] binding for `generate_device_keypair` uses
-// `serde-wasm-bindgen` and returns a plain JsValue (object), not a JSON
-// string. Calling JSON.parse on it throws `SyntaxError: "[object Object]"
-// is not valid JSON`. This regression test pins the contract.
+// The #[wasm_bindgen] binding for `register_device` uses `serde-wasm-bindgen`
+// and returns a plain JsValue (object), not a JSON string. Calling
+// JSON.parse on it would throw `SyntaxError: "[object Object]" is not
+// valid JSON`. This regression test pins that contract.
 
 describe('register_this_device', () => {
-  it('treats generate_device_keypair() as an object, not a JSON string', async () => {
+  it('treats register_device() return value as an object, not a JSON string', async () => {
     const state = makeState();
     state.gitHost = {} as never;
-    state.wasm.generate_device_keypair = () => ({
-      public_key_hex: 'aa'.repeat(32),
-      private_key_base64: 'AAAA',
+    state.wasm.register_device = () => ({
+      signing_public_key: 'aa'.repeat(32),
+      deploy_public_key: 'bb'.repeat(32),
     });
 
     vi.mocked(devices.addDevice).mockClear();