diff --git a/src/dsl/message.ts b/src/dsl/message.ts
index e8e6d18e6..0db80538a 100644
--- a/src/dsl/message.ts
+++ b/src/dsl/message.ts
@@ -85,9 +85,9 @@ export interface MessageProviders {
   [name: string]: MessageProvider;
 }
 
-export interface MessageStateHandlers {
+export interface MessageStateHandlers<T> {
   [name: string]: (
     state: string,
     params?: { [name: string]: string }
-  ) => Promise<unknown>;
+  ) => Promise<T>;
 }
diff --git a/src/dsl/options.ts b/src/dsl/options.ts
index 9ced9f790..0522270d3 100644
--- a/src/dsl/options.ts
+++ b/src/dsl/options.ts
@@ -69,7 +69,7 @@ export interface MessageProviderOptions {
   messageProviders: MessageProviders;
 
   // Prepare any provider states
-  stateHandlers?: MessageStateHandlers;
+  stateHandlers?: MessageStateHandlers<unknown>;
 }
 
 type ExcludedPactNodeVerifierKeys = Exclude<