Fix result type

ShallotJS · Apr 30, 2021 · eb07f31 · eb07f31
1 parent 587cad1
commit eb07f31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -1,4 +1,4 @@
-import type { APIGatewayProxyEvent, Handler } from 'aws-lambda';
+import type { APIGatewayProxyEvent, APIGatewayProxyResult, Handler } from 'aws-lambda';
 import type { TShallotErrorHandlerOptions } from '@shallot/http-error-handler/dist/aws';
 
 import ShallotAWS, { ShallotAWSHandler } from '@shallot/aws';
@@ -52,7 +52,7 @@ const ShallotSocketWrapper = <TEvent extends TShallotSocketEvent = TShallotSocke
     HttpErrorHandlerOpts?: TShallotErrorHandlerOptions;
     HttpJsonBodyParserOpts?: TShallotJSONBodyParserOptions;
   } = {}
-): ShallotAWSHandler<TEvent> => {
+): ShallotAWSHandler<TEvent, APIGatewayProxyResult> => {
   const wrappedResponseHandler: Handler = async (...args) => {
     const res = await handler(...args);
     return {