diff --git a/packages/ssi/src/index.ts b/packages/ssi/src/index.ts
index a2228a5..b04df54 100644
--- a/packages/ssi/src/index.ts
+++ b/packages/ssi/src/index.ts
@@ -62,7 +62,8 @@ import {
   JsonLdCredentialFormatService,
   ProofsModule,
   V2ProofProtocol,
-  ConnectionsModule
+  ConnectionsModule,
+  Agent
 } from '@aries-framework/core'
 import {
   GetCredentialsForRequestReturn,
@@ -124,7 +125,8 @@ export {
   JsonLdCredentialFormatService,
   ProofsModule,
   V2ProofProtocol,
-  ConnectionsModule
+  ConnectionsModule,
+  Agent
 }
 // Anoncreds
 export {
diff --git a/packages/ssi/src/providers/AgentProvider.tsx b/packages/ssi/src/providers/AgentProvider.tsx
index 8b4aa2a..fe1cb3b 100644
--- a/packages/ssi/src/providers/AgentProvider.tsx
+++ b/packages/ssi/src/providers/AgentProvider.tsx
@@ -1,10 +1,11 @@
 import type { Agent } from '@aries-framework/core'
+import type { AgentModulesInput } from '@aries-framework/core/build/agent/AgentModules'
 
 import AgentProvider, { useAgent } from '@aries-framework/react-hooks'
 
-const useAdeyaAgent: () => {
+const useAdeyaAgent: <AgentModules extends AgentModulesInput = AgentModulesInput>() => {
   loading: boolean
-  agent: Agent
+  agent: Agent<AgentModules>
 } = useAgent
 
 export { useAdeyaAgent, AgentProvider as AdeyaAgentProvider }