diff --git a/apps/ssi-abstraction/src/agent/__tests__/agent.controller.spec.ts b/apps/ssi-abstraction/src/agent/__tests__/agent.controller.spec.ts
index 5d24ea1a0a8e089e834ba6ae883f9c781bc82a3c..80a4c8065631d527966e316a6788d6eaedf4fb7a 100644
--- a/apps/ssi-abstraction/src/agent/__tests__/agent.controller.spec.ts
+++ b/apps/ssi-abstraction/src/agent/__tests__/agent.controller.spec.ts
@@ -24,7 +24,7 @@ describe('AgentController', () => {
       const result = { id: 'test' };
       jest
         .spyOn(agentService, 'getPublicDid')
-        .mockImplementation(() => Promise.resolve(result));
+        .mockResolvedValue(result)
 
       expect(await agentController.publicDid()).toBe(result);
     });