diff --git a/Sources/Vercel/Handlers/ExpressHandler.swift b/Sources/Vercel/Handlers/ExpressHandler.swift
index d206bd3..8b2f89c 100644
--- a/Sources/Vercel/Handlers/ExpressHandler.swift
+++ b/Sources/Vercel/Handlers/ExpressHandler.swift
@@ -5,6 +5,8 @@
 //  Created by Andrew Barba on 1/21/23.
 //
 
+import AWSLambdaRuntime
+
 public protocol ExpressHandler: RequestHandler {
 
     static var basePath: String { get }
@@ -18,7 +20,7 @@ extension ExpressHandler {
         return "/"
     }
 
-    public static func setup() async throws {
+    public static func setup(context: LambdaInitializationContext) async throws {
         // Create the router
         let router = Router(prefix: basePath)
         // Configure router in user code