From eb07f3136cefc2482229af219251e0ef445c8353 Mon Sep 17 00:00:00 2001
From: Bailey Tincher <btincher99@gmail.com>
Date: Thu, 29 Apr 2021 19:56:37 -0500
Subject: [PATCH] Fix result type

---
 src/index.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 1e709b1..79c7157 100644
--- 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 {